Senior Engineer, ML Training Platform

Summary

Join Motional as a Principal Engineer and lead the high-visibility ML Training Platform team. You will be responsible for designing, developing, and maintaining the robust systems that power our ML workflows. This critical role requires collaboration with infrastructure engineering, the ML team, and ML evaluation/metrics systems to accelerate training and evaluation processes. Success in this position demands understanding business priorities, thriving in ambiguous environments, and delivering models that improve on-road autonomous vehicle performance. The role involves building and maintaining scalable ML solutions in AWS using Kubernetes, implementing solutions in Python and PyTorch, and optimizing training infrastructure. The ideal candidate possesses strong Python coding skills and experience with AWS, Kubernetes, and popular ML frameworks.

Requirements

• 3+ years of strong Python coding skills
• Experience designing, developing and maintaining production systems in AWS cloud with Kubernetes
• Hands-on experience with popular ML frameworks (PyTorch or TensorFlow)
• Experience with CPU and GPU performance optimization
• Proven track record of operating highly-available systems at scale

Responsibilities

• Build and maintain scalable ML data processing, model training solutions in the AWS cloud infrastructure environment utilizing Kubernetes
• Implement solutions in our ML codebase (Python, Pytorch) and platform codebase (Volcano and Ray)
• Perform training and infrastructure performance optimization with various GPUs to improve model training speed and efficiency
• Communicate with machine learning engineers to identify obstacles and help prioritize solutions

Preferred Qualifications

• Hands-on experience with Ray in large scale environments
• Knowledge or hands on experience with Go
• Experience with ML data processing for large-scale deep learning training
• Experience working with large datasets and big data technologies

Benefits

• Medical
• Dental
• Vision
• 401k with a company match
• Health saving accounts
• Life insurance
• Pet insurance
• Bonus
• Company equity