Senior ML Infrastructure Engineer - Scale Training Platform

Aimlroles; Sunnyvale (CA); Hybrid; USD 170,000 - 241,000; Full time

Benefits offered by this job

Relocation benefits

GM benefits package

Job summary

General Motors is seeking an experienced Senior ML Engineer to design and build scalable ML training infrastructure that supports advanced AI research and model development for autonomous driving initiatives. You will collaborate with ML engineers, researchers, and partners on state-of-the-art AI solutions across GM vehicles.

The role emphasizes scalable training frameworks, observability, performance optimization, and cross-functional collaboration in a hybrid work environment with less than

Qualifications

• Bachelor's degree or higher in Computer Science or equivalent major or experience.
• 5+ years professional software engineering experience.
• 1+ years specialized experience in AI/ML infrastructure, enabling distributed training for scaling large ML models.
• Strong programming skills in Python with frameworks such as PyTorch (preferred) or TensorFlow.
• 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.

Responsibilities

• Design and development of scalable, reliable, high-performance ML framework to support model training at scale.
• Raise the bar on system observability, debuggability, and operational excellence, and user experience.
• Model training performance analysis and optimization to scale distributed training workflows and maximize resource utilization across heterogeneous hardware environments, and save cost.
• Collaborate with cross-functional teams to integrate new features and technologies into the platform.

Skills

Python

PyTorch

TensorFlow

Distributed computing

Cloud environments

GPU computing

Willing to travel

Education

Bachelor's degree in Computer Science

Tools

AWS

GCP

Azure

PyTorch 2.x+

Job description

General Motors is seeking an experienced Senior ML Engineer to design and build scalable ML training infrastructure that supports advanced AI research and model development for autonomous driving initiatives. You will collaborate with ML engineers, researchers, and partners on state-of-the-art AI solutions across GM vehicles.

The role emphasizes scalable training frameworks, observability, performance optimization, and cross-functional collaboration in a hybrid work environment with less than