Lead Machine Learning Engineer

Job Summary: We are on the lookout for a passionate and hands-on Lead Machine Learning Engineer with a strong focus on ML Training Infrastructure. In this pivotal role, you will craft and enhance scalable, reliable, and high-performance AI/ML platform infrastructure, empowering groundbreaking AI research and model development for intelligent driving technologies in General Motors vehicles. Your Responsibilities: Design and develop a robust ML framework that scales efficiently to support model training. Analyze and optimize model training performance, enhancing distributed training workflows while maximizing resource utilization across diverse hardware environments. Enhance system observability, debuggability, and operational excellence to improve user experience. Collaborate with cross-functional teams to integrate innovative features and technologies into the platform. Essential Skills & Qualifications: Bachelor's degree or higher in Computer Science or a related field, or equivalent experience. 3+ years of professional software engineering experience. 2+ years of specialized experience in AI/ML infrastructure, including enabling distributed training for large ML models. Proficiency in Python, with expertise in frameworks such as PyTorch (preferred), TensorFlow, and others. Familiarity with distributed computing, GPU computing, and cloud platforms (AWS, GCP, Azure). Willingness to travel to Sunnyvale, CA as necessary. Comfortable navigating ambiguous and dynamic environments. Preferred Qualifications: 5+ years of professional software engineering experience. Self-motivated, with a strong focus on driving results and delivering impact. In-depth knowledge and experience with PyTorch 2.x+ and distributed training frameworks. Experience in designing training frameworks that support FSDP, Pipeline Parallelism, and other advanced solutions for training large foundational models. Skilled in profiling, analyzing, debugging, and optimizing training and data loading performance. Excellent communication skills to facilitate resolution of debates, achieve consensus, communicate risks, and provide constructive feedback. Compensation: The compensation range for this position is between $170,000 and $240,000, depending on relevant factors. The role also includes incentive pay based on both individual and company performance. Relocation: This position may be eligible for relocation benefits. Benefits: GM offers a range of health and wellbeing programs, including medical, dental, vision plans, Health Savings Accounts, retirement savings plans, and more. This role is based remotely, but individuals living within a 50-mile radius of Mountain View, Sunnyvale, Detroit, Warren, or Milford are expected to report to one of these locations at least three times per week, depending on the manager's discretion. About GM: At General Motors, our mission is to achieve Zero Crashes, Zero Emissions, and Zero Congestion, and we are dedicated to leading the transformation needed to create a safer, more equitable world.

Created: 2026-03-04