Staff ML Engineer: Imitation Learning for Autonomous Driving

General Motors Company; United States; Hybrid; USD 189,000 - 280,000; Full time

Benefits offered by this job

Health and wellbeing benefits

GM vehicle program

Relocation assistance

Job summary

General Motors is seeking a Staff Software/ML Engineer to join the Imitation Learning (IL) team within Embodied AI Behavior Planning. You will build scalable ML software for training, evaluation, and deployment of learned driving policies, partnering with ML engineers to shorten the experimentation cycle while delivering production-quality systems.

The role emphasizes end-to-end pipelines, performance optimization on CPU/GPU, and integration with RL, simulation, and real-world driving scenarios.

Qualifications

• Master's or PhD in CS or related field.
• Track record leading complex software projects.
• Strong Python skills and maintainable code in large codebases.
• Hands-on PyTorch experience and ML training/evaluation systems.
• Experience profiling and optimizing ML pipelines (CPU/GPU, distributed).
• Ability to translate research needs into production software.

Responsibilities

• Architect and build modular ML software for data prep, training, evaluation, and production integration.
• Optimize end-to-end training/evaluation performance across CPU/GPU, data loading, and distribution.
• Enable scalable pretraining, fine-tuning, and RL integration workflows.
• Integrate closed-loop simulation and targeted scenarios for rapid feedback on models.
• Develop reusable model interfaces and data augmentation tools; ensure consistency across training and on-vehicle inference.
• Lead cross-team initiatives, mentor engineers, improve testing, reproducibility, and reliability.

Skills

Python programming

PyTorch ML frameworks

ML training/evaluation systems

Software architecture & leadership

Performance profiling & optimization

Production-grade software development

Education

Master's or PhD in Computer Science or related field

Tools

PyTorch

C++

GPU profiling tools

Job description

General Motors is seeking a Staff Software/ML Engineer to join the Imitation Learning (IL) team within Embodied AI Behavior Planning. You will build scalable ML software for training, evaluation, and deployment of learned driving policies, partnering with ML engineers to shorten the experimentation cycle while delivering production-quality systems.

The role emphasizes end-to-end pipelines, performance optimization on CPU/GPU, and integration with RL, simulation, and real-world driving scenarios.