Machine Learning Systems Engineer

Role Description

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.

What you'll be doing:

• 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.

Requirements

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.

Benefits

• Medical, dental, vision
• 401k with a company match
• Health saving accounts
• Life insurance
• Pet insurance
• And more

Salary Range

$144,000 — $192,000 USD

Company Description

Motional is a driverless technology company making autonomous vehicles a safe, reliable, and accessible reality. We’re driven by something more.

• Our journey is always people first.
• We aren't just developing driverless cars; we're creating safer roadways, more equitable transportation options, and making our communities better places to live, work, and connect.
• Our team is made up of engineers, researchers, innovators, dreamers and doers, who are creating a technology with the potential to transform the way we move.
• Higher purpose, greater impact.
• We’re creating first-of-its-kind technology that will transform transportation.
• Formed as a joint venture between Hyundai Motor Group and Aptiv, Motional is fundamentally changing how people move through their lives.
• Headquartered in Boston, Motional has operations in the U.S and Asia.