Jobs · Information Technology · Texas

Senior ML Infrastructure Engineer - Embodied AI Scaling Foundations

General Motors · Austin, TX · 1 mo ago
HybridInformation Technology$129k/yrFull-time

About the role

At General Motors, our product teams are redefining mobility through a human-centered design process, creating vehicles and experiences designed not just to be seen, but to be felt. We’re turning today’s impossible into tomorrow’s standard — from breakthrough hardware and battery systems to intuitive design, intelligent software, and next-generation safety and entertainment features. Every day, our products move millions of people as we aim to make driving safer, smarter, and more connected, shaping the future of transportation on a global scale.

As a Senior ML Infra Engineer on the Embodied AI Infra Foundation team, you will build the critical infrastructure that powers every machine learning engineer working on our cutting-edge Autonomous Driving models. From foundational models to state-of-the-art optimization, our work is at the heart of our mission. Our goal is simple: dramatically accelerate the machine learning development cycle, freeing our engineers to focus entirely on enhancing the safety and performance of the car, rather than managing infrastructure. We are committed to delivering products that are performant, easy to use, and exceptionally reliable. Your success will be measured by the success of our partner teams who rely on our robust systems to build the world's most advanced driverless vehicles.

Responsibilities

  • Lead the design, implementation, and deployment of scalable platforms and tools that drive machine learning model training and evaluation workflows across GM.
  • Own complex technical projects end-to-end, making key architectural decisions and technical trade-offs.
  • Be a core contributor to team planning, design reviews, and code quality.
  • Take a holistic view of projects, considering their impact across multiple teams, and proactively drive technical prioritization.
  • Collaborate closely with partner teams to ensure maximum benefit from the systems we build.
  • Help shape our team through technical interviewing with high, well-calibrated standards, and play an essential role in recruiting.
  • Mentor and onboard junior engineers and interns, helping them grow their careers.

Requirements

  • 3+ years of experience building large-scale distributed systems/applications or advanced ML applications.
  • Proven track record of building robust frameworks with high-quality, long-lasting APIs.
  • Deep understanding and practical experience with machine learning algorithms.
  • Expertise in building reliable, highly performant, and cost-efficient systems leveraging modern cloud infrastructure.
  • Hands-on experience with the entire ML development lifecycle and MLOps practices.
  • Demonstrated ability to collaborate effectively across multiple teams and organizations.
  • Proficiency working with containerization and orchestration technologies (Docker, Kubernetes).
  • A strong passion for self-driving technology and its transformative potential.
  • Exceptional coding skills in Python or C++.
  • BS, MS, or PhD in Computer Science, Math, or equivalent practical experience.

Qualifications

  • Experience with distributed training methodologies.
  • A background in optimizing model training performance.
  • Experience scaling model training across large clusters of GPUs/CPUs or other accelerators.
  • Familiarity with deep learning frameworks such as PyTorch, TensorFlow, etc.
  • A strong grasp of performance profiling and state-of-the-art training optimization algorithms, including their performance characteristics and effect on model convergence.
  • Experience with advanced build systems (Bazel, Buck, Blaze, or Cmake).
  • 3+ years of professional experience.

Schedule

This role is based remotely but, if you live within a 50-mile radius of an office, you are expected to report to that location three times a week, at minimum.

Pay

The salary range for this role is $128,700.00 to $261,300.00. The actual base salary a successful candidate will be offered within this range will vary based on factors relevant to the position. Bonus potential: an incentive pay program offers payouts based on company performance, job level, and individual performance.

Benefits

  • Medical, dental, and vision insurance
  • Health Savings Account and Flexible Spending Accounts
  • Retirement savings plan
  • Sickness and accident benefits, life insurance
  • Paid vacation and holidays
  • Tuition assistance programs
  • Employee assistance program
  • GM vehicle discounts

Similar jobs