Staff Software Engineer - Machine Learning

Job Description Role: The Smart Agents group is responsible for building the ML models and system to simulate road users in a variety of situations and generate the scenarios used for testing and training AV driving policies. If you think of Simulation as a video game our autonom ous vehicles train on to learn to drive, the Smart Agents team develops the ML/AI models that cont rol the other characters in the video game to interact in realistic ways as the av dr ives- eg , the other vehicles, bikers, and pedestrians. Our technology stack includes Generative AI models (GPT) and Reinforcement Learning (RL) policies . The Smart Agents group work closely with the rest of the Simulation, and our partners Behaviors, Perception, and Safety Engineers. The specific duties may include ML/RL model development as well as training loop development, streamlining optimization, integration, creating ML infrastructure , metrics, and data pipeline s for production model de plo yment as well as f or fast experimentation cycles. What You'll Do: • Support the team in developing machine learning (ML) and reinforcement learning (RL) models, including training loop development and optimization. • Streamline integration and create ML infrastructure, metrics, and data pipelines for production model deployment and rapid experimentation. • Work as part of an ML team and cont ri bute strong software engineering (SWE) expertise . • Support the ML team in accelerating project timelines, particularly in areas related to Autopilot, Lane Keep, and autonomous vehicle (AV) technologies. • Experience in simulation and robotics is highly desirable, with a preference for candidates from AV or robotics backgrounds rather than solely cloud-focused companies. Your Skills & Abilities: • 4+ years of experience in the field of robotics or latency-sensitive backend services • Background working with machine learning teams, algorithms, and models • Bonus: Experience building highly performant ML and system pipelines • Strong programming skills in modern C++ or Python Bonus: • Experience with profiling CPU and/or GPU software, process scheduling, and prioritization • Passionate about self-driving car technology and its impact on the world • Expertise in setting architectures that are scalable, efficient, fault-tolerant, and are easily extensible allowing for changes overtime without major disruptions. • Ability to design across multiple systems. Ability to both investigate in sophisticated areas as well as a good breadth of understanding of systems outside of your domain. • Ability to wear several hats shifting between coding, design, technical strategy, and mentorship combined with excellent judgment on when to switch cont exts to meet the greatest need. • Track record in deploying perception/prediction/av models into real world environments Your Skills & Abilities : • 4+ years of experience in the field of robotics or latency-sensitive backend services • Proven experience in machine learning and classification. Familiar with ML frameworks such as Tensorflow or PyTorch • Experience building highly performant ML and system pipelines • Strong programming skills in modern C++ or Python • Experience with profiling CPU and/or GPU software, process scheduling, and prioritization • Passionate about self-driving car technology and its impact on the world • Expertise in setting architectures that are scalable, efficient, fault-tolerant, and are easily extensible allowing for changes overtime without major disruptions. • Ability to design across multiple systems. Ability to both investigate in sophisticated areas as well as a good breadth of understanding of systems outside of your domain. • Ability to wear several hats shifting between coding, design, technical strategy, and mentorship combined with excellent judgment on when to switch contexts to meet the greatest need. • Track record in deploying perception/prediction/av models into real world environments • Experience working with RL and sequence prediction (ML) model s Compensation : The compensation information is a good faith estimate only. It is based on what a successful applicant might be paid in accordance with applicable state laws. The compensation may not be representative for positions located outside of New York, Colorado, California, or Washington. • The salary range for this role is $144,700 to $284,100. 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: GM offers a variety of health and wellbeing benefit programs. Benefit options include medical, dental, vision, Health Savings Account, Flexible Spending Accounts, retirement savings plan, sickness and accident benefits, life insurance, paid vacation & holidays, tuition assistance programs, employee assistance program, GM vehicle discounts and more. Remote: This role is based remotely but if you live within a 50-mile radius of Atlanta, Austin, Detroit, Warren, Milford or Mountain View , you are expected to report to that location three times a week, at minimum. Relocation: This job may be eligible for relocation benefits. #GM-AV-1 GM does not provide immigration-related sponsorship for this role. Do not apply for this role if you will need GM immigration sponsorship now or in the future. This includes direct company sponsorship, entry of GM as the immigration employer of record on a government form, and any work authorization requiring a written submission or other immigration support from the company (e.g., H1-B, OPT, STEM OPT, CPT, TN, J-1, etc). This role is categorized as hybrid. This means the selected candidate is expected to report to a specific location at least 3 times a week {or other frequency dictated by their manager}. About GM Our vision is a world with