Staff ML Infrastructure Engineer - Embodied AI Offboard Perception
Role details
Job location
Tech stack
Job description
As a Staff ML Infra Engineer on the Offboard Perception team within the Embodied AI organization, you will be a senior engineer responsible for developing and deploying offboard machine learning solutions that deliver ground-truth-quality world estimates for multiple partner teams, including onboard model teams, simulation, and evaluation. The models you build will influence every stage of autonomous vehicle development-from training and validation to testing and safety. You will work closely with cross-functional engineering teams, help shape technical direction in your domain, and support other engineers' growth through collaboration and mentorship. You will also help transition research into scalable onboard ML capabilities while continuously improving the autonomy stack.
What You'll Do
- Design, build, and maintain ML infrastructure that enables rapid development, training, evaluation, and deployment of offboard perception models.
- Own the integration of models into production systems, including packaging, validation, deployment, rollout strategies.
- Implement CI/CD pipelines for ML systems, including automated testing, model validation, performance regression checks, and deployment automation.
- Establish model evaluation and observability frameworks, including training metrics, inference performance metrics, data quality checks, and production monitoring dashboards.
- Develop infrastructure for experiment tracking and benchmarking, enabling teams to compare model architectures, datasets, hyperparameters, and training procedures in a reliable and repeatable way.
- Support efficient dataset curation and ingestion pipelines that help prioritize high-value data, accelerate iteration cycles, and improve model performance on hard-edge cases.
- Partner with ML engineers, researchers, and software teams to ensure models can be reliably integrated into larger autonomy stacks and production services at scale.
- Define and enforce best practices for ML systems engineering, including reproducibility, configuration management, artifact management, security, and operational readiness.
- Support technical collaboration through code reviews, design reviews, and mentorship, helping raise the quality and maintainability of ML infrastructure across the organization., Company Vehicle: 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.
Requirements
- Strong software engineering fundamentals, including experience building reliable, maintainable, and scalable production systems.
- Proficiency in Python, with experience using ML and scientific computing libraries such as PyTorch, NumPy, and related tooling.
- Experience building and supporting ML training and deployment pipelines, including data processing, experiment execution, model packaging, and production rollout.
- Experience deploying ML models into production environments, with understanding of end-to-end workflows such as validation, serving, monitoring, and lifecycle management.
- Familiarity with distributed training and large-scale compute infrastructure, including GPUs, cluster scheduling, and performance optimization for training workloads.
- Experience with containerization, orchestration, and automation tools such as Docker, Kubernetes, workflow schedulers, and CI/CD systems.
- Experience with model observability and operational metrics, including training metrics, inference performance, reliability monitoring, and data/model drift detection.
- Strong communication and collaboration skills, with the ability to work effectively across ML, software, data, and systems engineering teams.
- Experience in robotics, perception systems, or autonomous driving is preferred.
Remote/Hybrid: 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.
Benefits & conditions
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 the California Bay Area.
- The salary range for this role is $189,300.00 to $290,700.00. 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.
