Senior ML Training Infrastructure Architect

General Motors; Sunnyvale, Northern (CA, KY); Hybrid; USD 170,000 - 241,000; Full time

Benefits offered by this job

Medical

Dental

Vision

Health Savings Account

Job summary

General Motors is seeking a Senior ML Engineer to design and build scalable ML infrastructure that supports large-scale model training and AI research for intelligent driving technologies. You will work with ML engineers, researchers, and partners to deliver high-performance, observable, and cost-efficient training solutions.

The role requires strong Python expertise with PyTorch or TensorFlow, experience in distributed computing and cloud platforms, and willingness to travel to Sunnyvale, CA.

Qualifications

• Bachelors degree or higher in Computer Science or equivalent major OR equivalent relevant experience.
• 5+ years professional software engineering experience.
• 1+ 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.

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 solutions 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

Distributed training

Cloud environments

Communication

Ambiguity handling

Education

Bachelor's degree in Computer Science or equivalent

Tools

PyTorch

TensorFlow

Job description

General Motors is seeking a Senior ML Engineer to design and build scalable ML infrastructure that supports large-scale model training and AI research for intelligent driving technologies. You will work with ML engineers, researchers, and partners to deliver high-performance, observable, and cost-efficient training solutions.

The role requires strong Python expertise with PyTorch or TensorFlow, experience in distributed computing and cloud platforms, and willingness to travel to Sunnyvale, CA.