Staff Software Engineer - Machine Learning

Position Overview

The Smart Agents group is responsible for building ML models and systems to simulate road users and generate realistic testing scenarios for autonomous vehicle (AV) driving policies. Working closely with Simulation, Behaviors, Perception, and Safety Engineers, this role involves developing ML/RL models, training loops, integration pipelines, and optimization strategies for both production deployments and rapid experimentation.

Key Responsibilities

• Support the development of machine learning (ML) and reinforcement learning (RL) models, including training loop creation and optimization.
• Streamline system integration and develop ML infrastructure, metrics, and data pipelines for both production deployment and fast experimentation cycles.
• Collaborate with the ML team by contributing strong software engineering expertise.
• Accelerate project timelines in areas related to Autopilot, Lane Keep, and other AV technologies.
• Leverage simulation and robotics experience with a preference for candidates from AV or robotics backgrounds.

Required Qualifications

• 4+ years of experience in robotics or latency-sensitive backend services.
• Proven experience in machine learning and classification, with familiarity using ML frameworks such as TensorFlow or PyTorch.
• Strong programming skills in modern C++ or Python.
• Experience building highly performant ML and system pipelines.
• Experience with profiling CPU/GPU software, process scheduling, and prioritization.
• Demonstrated experience working with RL and sequence prediction (ML) models.

Preferred Qualifications

• Passion for self-driving car technology and its impact on the world.
• Expertise in designing scalable, efficient, and fault-tolerant architectures.
• Ability to design across multiple systems while shifting between coding, design, technical strategy, and mentorship.
• Track record in deploying perception, prediction, or AV models into real-world environments.

Benefits & Perks

• Compensation: Salary range is $134,000 to $235,900 with the actual base salary determined by relevant factors.
• Bonus Potential: Incentive pay program based on company performance, job level, and individual contributions.
• Benefits: GM provides a variety of health and wellbeing programs including medical, dental, vision, Health Savings Account, Flexible Spending Accounts, retirement savings, sickness and accident benefits, life insurance, paid vacation & holidays, tuition assistance, an employee assistance program, GM vehicle discounts, and more.
• Remote work is available; however, candidates residing within a 50-mile radius of Atlanta, Austin, Detroit, Warren, Milford, or Mountain View are expected to report on-site at least three times a week.
• Relocation benefits may be eligible.