Machine Learning Engineer, AML - Engine (Multiple )

Machine Learning Engineer, AML - Engine

Location: San Jose

Team: Technology

Employment Type: Regular

Job Code: A22474A

Responsibilities

Apply machine learning and robotics research knowledge including LLM and reinforcement learning to analyze and optimize deep learning model architectures. Design and implement high-performance CUDA kernels to accelerate model inference and computation-heavy operators. Analyze computation graphs to identify optimization opportunities across different model structures. Leverage parallel computing techniques such as CUDA and OpenMP to improve throughput, latency, and resource utilization in ML workloads. Migrate between TensorFlow and PyTorch model serving frameworks, ensuring correctness and performance consistency. Develop algorithm-level optimizations for both in-graph and out-of-graph computations, balancing AUC and execution efficiency.

Qualifications

Must have a Master's degree or foreign equivalent degree in Computer Science, Engineering (any), Information Technology, Data Science, Robotics, Mathematics, or a related quantitative field, and 1 year of related work experience; OR a Bachelor's degree or foreign equivalent degree in Computer Science, Engineering (any), Information Technology, Data Science, Robotics, Mathematics, or a related quantitative field, and 3 years of related work experience. Of the required experience, must have 1 year of experience in each of the following: Coding using Python or C++; Optimize deep learning training and inference pipelines on GPU and CPU architectures; Implement CUDA-based performance optimizations for machine learning workloads; Improve systems-level performance across machine learning frameworks; Developing and deploying large-scale machine learning systems; and Utilizing Big data framework using MySQL, Spark, Hadoop, or Flink.

Type: Full time, 40 hours/week Location: San Jose, CA Salary Range: $187741 - $316800 per year

To Apply, click the apply button below. Contact if you have difficulty submitting resume through the website. ByteDance is committed to creating an inclusive space where employees are valued for their skills, experiences, and unique perspectives. Our platform connects people from across the globe and so does our workplace. At ByteDance, our mission is to inspire creativity and enrich life. To achieve that goal, we are committed to celebrating our diverse voices and to creating an environment that reflects the many communities we reach. We are passionate about this and hope you are too. ByteDance is committed to providing reasonable accommodations in our recruitment processes for candidates with disabilities, pregnancy, sincerely held religious beliefs or other reasons protected by applicable laws. If you need assistance or a reasonable accommodation, please reach out to us at #IND-DNI