SIGNUP

Job Opportunities

Find jobs near you, whether onsite, hybrid, or remote.
  • Similar Jobs to: Machine Learning Systems Engineer
XX
Machine Learning Systems EngineerMotional AD Inc. • San Francisco, California, United States
Apply Now
XX

Machine Learning Systems Engineer

Motional AD Inc.
  • US
    San Francisco, California, United States
  • US
    San Francisco, California, United States
Apply Now

About

Position Summary We are looking for a Machine Learning Systems Engineer to join our ML Acceleration team. In this role, you will be responsible for the core systems that enable our researchers to train frontier models at scale, focusing obsessively on speed, cost, reliability, and throughput. You will work at the intersection of machine learning research and high-performance systems engineering. Your work will directly impact our ability to scale large-scale distributed model training and reduce the time-to-convergence for our next generation of models.
Responsibilities
Performance Profiling & Optimization : Utilize profiling tools (e.g., Nsight, PyTorch Profiler) to identify bottlenecks in data loading, gradient computation, and communication. Implement optimizations like kernel fusion, sharding, and tiling to improve step time.
Distributed Training : Optimize distributed training pipelines using frameworks such as PyTorch Distributed.
Kernel Development : Design and maintain high-performance GPU kernels in Triton or CUDA for state-of-the-art ML workloads.
Data Pipeline Engineering : Optimize robust data loading pipelines that maximize training throughput.
Qualifications
Education : Bachelor’s, Master’s degree, or PhD in Computer Science, Computer Engineering, or a related technical discipline.
Software Engineering : Strong proficiency in Python.
ML Frameworks : Extensive hands-on experience with PyTorch.
ML Knowledge : Experience optimizing machine learning model execution during training and inference, alongside a strong understanding of fundamental machine learning concepts, architectures, and processes.
Problem Solving : Exceptional analytical and problem-solving skills, with a bias for action and a data-driven approach to technical challenges.
We encourage a hybrid schedule with in-office time at one of our locations in Boston, Pittsburgh, or Las Vegas to support collaboration, or this role can be fully remote.
Locations: Boston, Las Vegas, Pittsburgh, Remote U.S. Only
The salary range for this role is an estimate based on a wide range of compensation factors including but not limited to specific skills, experience and expertise, role location, certifications, licenses, and business needs. The estimated compensation range listed in this job posting reflects base salary only. This role may include additional forms of compensation such as a bonus or company equity.
Salary Range: $144,000 — $192,000 USD
Candidates for certain positions are eligible to participate in Motional’s benefits program. Motional’s benefits include but are not limited to medical, dental, vision, 401k with a company match, health saving accounts, life insurance, pet insurance, and more.
Motional is an Equal Opportunity Employer. We celebrate diversity and are committed to creating an inclusive environment for all employees. To comply with Federal Law, we participate in E-Verify. All newly-hired employees are queried through this electronic system established by the DHS and the SSA to verify their identity and employment eligibility.
#J-18808-Ljbffr
  • San Francisco, California, United States

Languages

  • English
Notice for Users

This job comes from a TieTalent partner platform. Click "Apply Now" to submit your application directly on their site.