Staff Software Engineer, Imitation Learning - Behavior Planning, Embodied AI
General Motors2 days ago
Base Salary
$189k - $280k/yr
Responsibilities
- Architect and build modular ML software covering data preparation, training, evaluation, and production integration.
- Optimize ML training and evaluation pipelines, including data loading, CPU/GPU utilization, distributed execution, and workflow orchestration.
- Enable reusable imitation-learning pretraining, fine-tuning, post-training, and reinforcement-learning workflows.
- Integrate open-loop behavioral tests, closed-loop simulation, and targeted scenarios for rapid model-quality feedback.
- Build reusable model interfaces and data-augmentation tools while maintaining training, simulation, and on-vehicle inference consistency.
- Lead cross-team technical initiatives, mentor engineers, and improve software modularity, testing, reproducibility, and reliability.
Requirements
- Master’s or PhD in Computer Science or a related field, or equivalent practical experience.
- Track record 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 involving data processing, CPU/GPU execution, and distributed training.
- Strong technical judgment and ability to translate research needs into reliable production software.
- Preferred qualifications include experience in autonomous driving, robotics, behavior planning, trajectory prediction, imitation learning, reinforcement learning, multistage training, distillation, synthetic data generation, closed-loop simulation, model deployment, training-inference consistency testing, modern C++, GPU profiling, model compilation, or ML runtime optimization.
Benefits
- Fully remote or hybrid work arrangement.
- Medical, dental, vision, Health Savings Account, Flexible Spending Accounts, retirement savings plan, sickness and accident benefits, life insurance, paid vacation and holidays, tuition assistance, employee assistance program, and GM vehicle discounts.
- Eligibility for the company vehicle evaluation program after successful motor vehicle report review; participants generally must purchase or lease a qualifying GM vehicle every four years unless an exception applies.
- The role may be eligible for relocation benefits.
- Bonus potential through an incentive pay program based on company performance, job level, and individual performance.
Categories
About General Motors
General Motors designs, manufactures, and sells cars, trucks, and electric vehicles for consumers and commercial fleets under brands including Chevrolet, GMC, Cadillac, and Buick. A public company on the NYSE headquartered in Detroit and founded in 1908, it operates globally and is developing EVs on its Ultium battery platform. Revenue comes from vehicle and parts sales, connected services such as OnStar, and financing through GM Financial.