Note: The job is a remote job and is open to candidates in USA. General Motors is seeking an experienced Staff Machine Learning Engineer to lead the design and development of scalable AI/ML platform infrastructure. The role involves driving technical direction, collaborating with teams, and optimizing model training performance to enhance intelligent driving technologies.
Responsibilities
- Define and drive the architecture, design, and development of scalable, reliable, and high-performance ML frameworks and platform capabilities to support model training at scale
- Lead model training performance analysis and optimization efforts across distributed training workflows, improving scalability, efficiency, and cost across heterogeneous hardware environments
- Raise the bar on system observability, debuggability, operational excellence, and developer experience across the ML training stack
- Own large, ambiguous, cross-functional technical initiatives from strategy through execution, including technical roadmap definition, tradeoff analysis, and delivery
- Influence platform direction by identifying long-term infrastructure investments, setting engineering standards, and driving adoption of best practices across teams
- Collaborate across organizational boundaries to align requirements, resolve technical disagreements, and integrate new capabilities into the platform ecosystem
- Mentor engineers through design reviews, technical guidance, and hands-on partnership, while elevating engineering quality across the team
Skills
- Bachelor's degree or higher in Computer Science or a related field, or equivalent practical experience
- 7+ years of professional software engineering experience
- 5+ years of specialized experience in AI/ML infrastructure, such as enabling distributed training for large-scale ML models
- Strong programming skills in Python, with deep proficiency in frameworks such as PyTorch (preferred), TensorFlow, or similar ML systems
- Proven experience designing and operating distributed systems for ML training, including distributed computing, GPU computing, and cloud environments (AWS, GCP, Azure)
- Demonstrated track record of leading technically ambiguous, cross-team infrastructure initiatives and driving them to measurable impact
- Strong architectural judgment and ability to make sound technical tradeoffs across performance, reliability, usability, and cost
- Willingness to travel to Sunnyvale, CA as needed
- Comfortable operating in highly ambiguous and dynamic environments
- 7+ years of professional software engineering experience
- Deep expertise in PyTorch 2.x+ and distributed training frameworks
- Experience designing and developing training platforms that support FSDP, pipeline parallelism, and other scalable solutions for training large foundational models
- Experience profiling, analyzing, debugging, and optimizing training and data loading performance at scale
- Strong record of technical leadership through architecture reviews, roadmap influence, and cross-team execution
- Excellent communication skills, with the ability to build consensus, navigate controversial decisions, communicate risks clearly, and provide constructive technical feedback
- Self-motivated, execution-oriented, and motivated by delivering broad organizational impact
Benefits
- Bonus Potential: An incentive pay program offers payouts based on company performance, job level, and individual performance.
- Relocation: This job may be eligible for relocation benefits.
- GM offers a variety of health and wellbeing benefit programs. Benefit options include medical, dental, vision, Health Savings Account, Flexible Spending Accounts, retirement savings plan, sickness and accident benefits, life insurance, paid vacation & holidays, tuition assistance programs, employee assistance program, GM vehicle discounts and more.
- Upon successful completion of a motor vehicle report review, you will be eligible to participate in a company vehicle evaluation program, through which you will be assigned a General Motors vehicle to drive and evaluate. Note: program participants are required to purchase/lease a qualifying GM vehicle every four years unless one of a limited number of exceptions applies.
- This role is categorized as remote. This means the selected candidate may be based anywhere in the country of work and is not expected to report to a GM worksite unless directed by their manager.
- From day one, we're looking out for your well-being–at work and at home–so you can focus on realizing your ambitions.
- General Motors offers opportunities to all job seekers including individuals with disabilities. If you need a reasonable accommodation to assist with your job search or application for employment, email us or call us at 1-800-865-7580. In your email, please include a description of the specific accommodation you are requesting as well as the job title and requisition number of the position for which you are applying.
Company Overview
Company H1B Sponsorship