ML Systems Engineer - High-Performance Distributed Training

Motional; Pittsburgh (Allegheny County); Hybrid; USD 144,000 - 192,000; Full time

Benefits offered by this job

Medical benefits

Dental and vision insurance

401k with company match

Pet insurance

Health savings accounts

Job summary

Motional in Pittsburgh is seeking a Machine Learning Systems Engineer to enhance the systems that support model training. You will leverage your expertise in Python and PyTorch to drive optimizations and handle performance profiling.

This role supports a hybrid work environment, encouraging collaboration across teams. Ideal candidates should possess a degree in computer science or related fields and a strong problem-solving mindset. The position offers a competitive salary range, starting from $144,000 to $192,000.

Qualifications

• Bachelor’s, Master’s degree, or PhD in Computer Science, Computer Engineering, or related technical discipline.
• Strong proficiency in Python.
• Extensive hands-on experience with PyTorch.
• Experience optimizing machine learning model execution during training and inference.

Responsibilities

• Utilize profiling tools to identify bottlenecks in data loading, gradient computation, and communication.
• Optimize distributed training pipelines using frameworks such as PyTorch Distributed.
• Design and maintain high-performance GPU kernels in Triton or CUDA.

Skills

Proficiency in Python

Hands-on experience with PyTorch

Analytical and problem-solving skills

Education

Bachelor’s, Master’s, or PhD in Computer Science or related field

Job description

Motional in Pittsburgh is seeking a Machine Learning Systems Engineer to enhance the systems that support model training. You will leverage your expertise in Python and PyTorch to drive optimizations and handle performance profiling.

This role supports a hybrid work environment, encouraging collaboration across teams. Ideal candidates should possess a degree in computer science or related fields and a strong problem-solving mindset. The position offers a competitive salary range, starting from $144,000 to $192,000.