ML Systems Engineer - High-Performance Distributed Training

Motional; Boston (MA); Hybrid; USD 144,000 - 192,000; Full time

Benefits offered by this job

Medical, dental, and vision insurance

401k with company match

Life insurance

Pet insurance

Job summary

Motional is seeking a Machine Learning Systems Engineer to join our ML Acceleration team in Boston. In this role, you'll enhance core systems enabling researchers to train models effectively by optimizing performance, reliability, and throughput.

The ideal candidate will possess a degree in Computer Science, with strong Python and hands-on experience in PyTorch. This position allows for a hybrid work schedule, furthering collaboration while supporting a flexible work environment.

Qualifications

• Strong proficiency in Python programming language.
• Experience with PyTorch and optimizing model execution.
• Bachelor's, Master's, or PhD in Computer Science or related field.

Responsibilities

• Utilize profiling tools to identify performance bottlenecks.
• Optimize distributed training pipelines with PyTorch frameworks.
• Develop high-performance GPU kernels for ML workloads.

Skills

Python

Performance Profiling

Distributed Training

Machine Learning Concepts

Problem Solving

Education

Bachelor’s, Master’s degree, or PhD in Computer Science or related discipline

Tools

PyTorch

Nsight

Triton

CUDA

Job description

Motional is seeking a Machine Learning Systems Engineer to join our ML Acceleration team in Boston. In this role, you'll enhance core systems enabling researchers to train models effectively by optimizing performance, reliability, and throughput.

The ideal candidate will possess a degree in Computer Science, with strong Python and hands-on experience in PyTorch. This position allows for a hybrid work schedule, furthering collaboration while supporting a flexible work environment.