Motional

Machine Learning Systems Engineer

Pittsburgh, Pennsylvania, United States

Sponsorship not specified$144k-$192kDetected 28 days ago

Machine LearningPyTorchSystems EngineeringResearchCollaborationProblem SolvingPipeline Integrity

About the role

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.

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.
Design and maintain high-performance GPU kernels in Triton or CUDA for state-of-the-art ML workloads.
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.
To do so successfully, we must design for everyone in our cities and on our roads.
We believe in building a great place to work through a progressive, global culture that is diverse, inclusive, and ensures people feel valued at every level of the organization.

Utilize profiling tools (e.g., Nsight, PyTorch Profiler) to identify bottlenecks in data loading, gradient computation, and communication.

$144,000 - $192,000 USD
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.
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.
The recruiter assigned to this role can share more information about the specific compensation and benefit details associated with this role during the hiring process.
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.

This listing is sourced directly from Motional's careers page and normalized into a canonical job model.