General Motors · 5 hours ago
Staff Machine Learning Engineer, AI Platform
Sunnyvale, CA
Full-time
Hybrid
Senior Level, Lead/Staff
$165K/yr - $291K/yr
7+ years expGeneral Motors is seeking an experienced Staff Machine Learning Engineer to design and build scalable AI/ML platform infrastructure. The role involves collaborating with machine learning engineers and research scientists to develop advanced AI solutions for intelligent driving technologies.
AutomotiveElectric VehicleInformation ServicesManufacturingTransportation
Responsibilities
Lead the design and development of scalable, reliabile, high-performance ML framework to support model training at scale
Lead model training performance analysis and optimizaiton solutions to scale distributed training workflows and maximize resource utilization across heterogeneous hardware environments, and save cost
Raise the bar on system observability, debuggability, and operational excellence, and user experience
Collaborate with cross-functional teams to integrate new features and technologies into the platform
Qualification
Required
Bachelors or higher degree in Computer Science or equivalent major or equivalent experience
7+ years professional software engineering experience
3+ years specialized experience in AI/ML infrastructure, e.g., enabling distributed training for scaling large ML models
Strong programming skills in Python, with proficiency in frameworks such as, PyTorch (preferred), TensorFlow, or similar
Experience with distributed computing, GPU computing, and cloud environments (AWS, GCP, Azure)
Willingness to travel to Sunnyvale, CA as needed
Comfortable working in highly ambiguous and dynamic environments
Preferred
Self-motivated, strong execution, impact-delivering oriented
Extensive knowledge and experience with PyTorch 2.x+ and distributed training framework
Experience with design and development of training framework that supports FSDP, Pipeline Parallelism and other scalable solutions to training large foundational models
Experience with profiling, analysis, debugging and optimizing training and dataloading performance
Excellent communication skills to resolve controversial, make consensus, communicate risks and give constructive feedback
Benefits
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
Company
General Motors
General Motors is an automotive company that designs, produces, markets, and distributes vehicles and vehicle parts.
10001+ employees
H1B Sponsorship
General Motors has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (787)
2024 (740)
2023 (450)
2022 (795)
2021 (748)
2020 (452)
Funding
Current Stage
Public CompanyTotal Funding
$8.51BKey Investors
US Department of Energy
2025-05-05Post Ipo Debt· $2B
2024-10-31Grant· $8M
2024-07-11Grant· $500M
Leadership Team
Mary Barra
Chair and Chief Executive Officer
Tony Cervone
Senior Vice President, Global Communications
Recent News
2026-01-12
SlashGear.com
2026-01-12
2026-01-12
Company data provided by crunchbase