Staff AI / ML Engineer - Embodied AI

About the Role

As a Staff AI/ML Engineer in the Embodied AI organization, you will serve as a senior individual contributor developing and deploying machine learning solutions that directly impact autonomous driving performance. You will design and implement advanced ML models that enable safe and reliable vehicle behavior across diverse real-world scenarios. In this role, you will partner closely with cross-functional engineering teams, contribute to core technical direction within your domain, and support the growth of engineers through technical collaboration and mentorship. You will help translate research into scalable onboard ML solutions while contributing to the continuous improvement of our autonomy stack.

Responsibilities

• Design and implement ML solutions aligned with GM’s autonomous driving objectives.
• Apply techniques such as unsupervised pre-training, imitation learning, reinforcement learning, model scaling/selection, foundation modeling, to solve problems in object detection/tracking/classification, trajectory generation, and safe AI.
• Collaborate with cross-functional teams to deploy models and algorithms into onboard driving systems.
• Contribute to applied research efforts and remain current with advancements in ML frameworks and methods.
• Design and build efficient infrastructure, pipelines, and tooling to facilitate fast-paced model iterations.
• Drive technical execution from prototyping through production deployment, documenting learnings and best practices.
• Support and mentor engineers through technical collaboration and code reviews, fostering knowledge sharing and engineering excellence.

Qualifications

• Bachelor’s, Master’s, or PhD in Computer Science, Robotics, Machine Learning, or related field.
• Experience working with large-scale foundation models and alignment methods applied to real-world systems.
• Demonstrated ability to deliver applied ML solutions under real-world constraints and timelines.
• Proficiency in PyTorch and Python.
• Experience building and scaling model training pipelines enabling efficient iteration across teams.
• Strong data processing skills using tools such as NumPy, Pandas, and Apache Spark.
• Strong communication skills enabling effective collaboration across engineering teams.
• Experience deploying ML models into production environments and understanding end-to-end deployment workflows.
• Experience in robotics or autonomous driving systems preferred.

Schedule

This role is based remotely but if you live within a 50-mile radius of Austin, Detroit, Warren, Milford or Mountain View, you are expected to report to that location three times per week, at minimum.

Pay

The salary range for this role is $218,800.00 to $335,300.00. The actual base salary a successful candidate will be offered within this range will vary based on factors relevant to the position. 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
• Relocation benefits (if eligible)

Upon successful completion of a motor vehicle report review, you will be eligible to participate in a company vehicle evaluation program, through which you will be assigned a General Motors vehicle to drive and evaluate. Note: program participants are required to purchase/lease a qualifying GM vehicle every four years unless one of a limited number of exceptions applies.