Staff Software Engineer, Imitation Learning – Behavior Planning, Embodied AI

General Motors

. Architect and build modular machine-learning software spanning data preparation, training, evaluation, and production integration .

Posted 11/11/2026full-timeUnited StatesLead💰 $189,000 - $280,000 per yearWebsite

Core Competencies

Role fit

Core Competencies

Use this summary to align your resume positioning with the role.

Demonstrates expertise in architecting and building modular machine-learning software, with a strong focus on optimizing performance across training, evaluation, and deployment. Proficient in Python and frameworks like PyTorch, with a solid understanding of reinforcement learning and autonomous driving applications.

Highest-signal resume keywords

Python ProgrammingPyTorch FrameworkMachine Learning OptimizationSoftware Architecture LeadershipAutonomous Driving Experience

ATS Keywords

Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills

Machine Learning Software DevelopmentData PreparationModel EvaluationPerformance ProfilingDistributed TrainingReinforcement LearningImitation LearningData ProcessingModel DeploymentC++ Programming

Soft Skills

Technical JudgmentMentoringCross-Team Collaboration

Tools & Technologies

GPU Profiling ToolsML Training SystemsClosed-Loop Simulation

Certifications & Qualifications

Master’s Degree in Computer SciencePhD in Computer Science

Industry Keywords

Autonomous DrivingRoboticsBehavior PlanningTrajectory PredictionSynthetic Data Generation

Tech Stack

Tools & technologies

PythonPyTorchC++

About the role

Key responsibilities & impact

Architect and build modular machine-learning software spanning data preparation, training, evaluation, and production integration
Optimize end-to-end training and evaluation performance, including data loading, CPU/GPU utilization, distributed execution, and workflow orchestration
Enable reusable imitation-learning pretraining and rapid fine-tuning and post-training workflows, including reinforcement-learning integration
Integrate open-loop behavioral tests, closed-loop simulation, and targeted scenarios to provide fast, actionable model-quality feedback
Build reusable model interfaces and data-augmentation tools
Maintain consistency across training, simulation, and on-vehicle inference
Lead cross-team technical initiatives
Mentor engineers
Improve software modularity, testing, reproducibility, and reliability
Partner with ML engineers to develop scalable training, evaluation, and deployment solutions for learned driving policies

Requirements

What you’ll need

Master’s or PhD degree in Computer Science, a related field, or equivalent practical experience
Track record of leading complex software projects, setting architecture, and delivering hands-on implementation across team boundaries
Strong Python skills and experience building maintainable, well-tested software in large codebases
Hands-on experience with PyTorch or comparable frameworks and ML training or evaluation systems
Experience profiling and optimizing performance-sensitive ML pipelines, including data processing, CPU/GPU execution, and distributed training
Strong technical judgment and ability to translate research needs into reliable production software
Preferred: Experience in autonomous driving, robotics, behavior planning, or trajectory prediction
Preferred: Familiarity with imitation learning, reinforcement learning, multistage training, distillation, or synthetic data generation
Preferred: Experience with closed-loop simulation, model deployment, and training–inference consistency testing
Preferred: Modern C++ skills or experience with GPU profiling, model compilation, and ML runtime optimization

Benefits

Comp & perks

Incentive pay program with payouts based on company performance, job level, and individual performance
Medical, dental, and vision benefits
Health Savings Account
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
Company vehicle evaluation program, subject to motor vehicle report review and program conditions
Potential relocation benefits