General Motors · 3 days ago
Senior Machine Learning Engineer
Home, KS
Full-time
Hybrid
Senior Level
$134K/yr - $236K/yr
5+ years expGeneral Motors is seeking an experienced engineer in ML Training Infrastructure to design and build scalable AI/ML platform infrastructure. The role involves collaborating with machine learning engineers and research scientists to develop AI solutions for intelligent driving technologies.
AutomotiveElectric VehicleInformation ServicesManufacturingTransportation
Responsibilities
Participate in the design and development of scalable, reliable, high-performance ML framework to support model training at scale
Participate in model training performance analysis and optimization 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
5+ years professional software engineering experience
2+ 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
Experience with Apache Parquet, Apache Arrow, Ray, Ray Data
Strong programming skills in C++
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
Dallas Morning News
2026-01-07
DBusiness Magazine
2026-01-07
Digital Journal
2026-01-06
Company data provided by crunchbase