Senior Machine Learning Engineer - ML Training Infrastructure
General Motors5 days ago
Base Salary
$170k - $241k/yr
Responsibilities
- Design and develop scalable ML frameworks for large-scale model training.
- Analyze and optimize distributed training performance, resource utilization, cost, and data-loading performance across heterogeneous hardware.
- Improve system observability, debuggability, operational excellence, and user experience.
- Collaborate with machine learning engineers, research scientists, and cross-functional partners to integrate new platform features and technologies.
Requirements
- Bachelor's degree or higher in Computer Science or an equivalent major, or equivalent relevant experience.
- 5+ years of professional software engineering experience.
- At least 1 year of specialized AI/ML infrastructure experience, such as enabling distributed training for large ML models.
- Strong Python programming skills and proficiency with PyTorch, TensorFlow, or similar frameworks.
- Experience with distributed computing, GPU computing, and AWS, GCP, or Azure cloud environments.
- Preferred qualifications include extensive PyTorch 2.x+ and distributed-training experience, FSDP and pipeline parallelism experience, and experience optimizing training and data-loading performance.
- Willingness to travel to Sunnyvale, California as needed and ability to work in ambiguous, dynamic environments.
Benefits
- Health, dental, vision, Health Savings Account, Flexible Spending Account, retirement savings, sickness and accident benefits, life insurance, paid vacation and holidays, tuition assistance, employee assistance, and GM vehicle discounts.
- Hybrid work arrangement requiring on-site reporting at least 3 times per week or at a frequency set by the manager.
- Travel required at less than 25%.
- Potential eligibility for relocation benefits.
Tech Stack
Categories
About General Motors
General Motors designs, manufactures, and sells cars, trucks, and electric vehicles for consumers and commercial fleets under brands including Chevrolet, GMC, Cadillac, and Buick. A public company on the NYSE headquartered in Detroit and founded in 1908, it operates globally and is developing EVs on its Ultium battery platform. Revenue comes from vehicle and parts sales, connected services such as OnStar, and financing through GM Financial.