Remote ML Systems Engineer — Distributed Training & GPU

Motional AD Inc.; United States; Hybrid; USD 110,000 - 170,000; Full time

Job summary

Motional AD Inc. seeks a full-time Machine Learning Systems Engineer to optimize high-performance ML systems.

The role focuses on enhancing distributed training pipelines, developing efficient GPU kernels, and improving data loading, with the option to work remotely or in-office locations in Boston, Pittsburgh, or Las Vegas. The ideal candidate has advanced degrees in a related field, strong Python skills, and extensive PyTorch experience to push the performance of ML workloads during training

Qualifications

• Bachelor's, Master's, or PhD in CS/CE or related field.
• Strong proficiency in Python.
• Extensive hands-on experience with PyTorch.
• Experience optimizing ML model training and inference.
• Analytical skills with a data-driven approach to problem-solving.

Responsibilities

• Profile ML systems to identify performance bottlenecks.
• Optimize distributed training pipelines using PyTorch Distributed.
• Design and maintain high-performance GPU kernels for ML workloads.

Skills

Python programming

Analytical skills

Data-driven problem solving

ML system optimization

Education

Bachelor's/Master's/PhD in CS/CE or related

Tools

PyTorch

Profiling tools

Job description

Motional AD Inc. seeks a full-time Machine Learning Systems Engineer to optimize high-performance ML systems.

The role focuses on enhancing distributed training pipelines, developing efficient GPU kernels, and improving data loading, with the option to work remotely or in-office locations in Boston, Pittsburgh, or Las Vegas. The ideal candidate has advanced degrees in a related field, strong Python skills, and extensive PyTorch experience to push the performance of ML workloads during training