Senior Machine Learning Engineer

General Motors invites you to join as a Senior Machine Learning Engineer. This role focuses on designing and building scalable ML training infrastructure that underpins AI research and model development, enabling intelligent driving technologies across GM vehicles. You will collaborate with ML engineers and researchers to deliver a scalable, reliable platform with strong performance, observability, and cross-functional collaboration. The team emphasizes rigorous engineering, a collaborative culture, and opportunities to impact the future of mobility. A comprehensive rewards package supports health, growth, and work-life balance.

Responsibilities

• Design and develop a scalable, reliable, high-performance ML framework to support model training at scale.
• Analyze and optimize model training performance to scale distributed workflows and maximize resource utilization across heterogeneous hardware, while controlling costs.
• Raise the bar on system observability, debuggability, operational excellence, and user experience.
• Collaborate with cross-functional teams to integrate new features and technologies into the platform.

Requirements

• Bachelors degree or higher in Computer Science or an equivalent major, or relevant experience.
• 3+ years of professional software engineering experience.
• 2+ years of specialized experience in AI/ML infrastructure, enabling distributed training for scaling large ML models.
• Strong Python programming skills, with proficiency in PyTorch (preferred), TensorFlow, or similar frameworks.
• 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.

Technologies

• Python
• PyTorch
• TensorFlow
• AWS
• GCP
• Azure
• PyTorch 2.x+
• FSDP
• Pipeline Parallelism

Compensation

• Salary range: $170,000 to $240,000 per year. The actual base salary offered within this range will vary based on factors relevant to the position.
• Bonus Potential: An incentive pay program pays out based on company performance, role level, and individual performance.

Location and Relocation

Sunnyvale, CA (remote)

• This role may be eligible for relocation benefits.

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