# Senior/Staff Software Engineer, ML Infrastructure, Optimization

**Company:** [Nuro](https://hotfix.jobs/companies/nuro)
**Location:** Mountain View, CA
**Role:** ML Engineering
**Salary:** $194k – $352k/yr
**Experience:** 3+ years
**Skills:** Python, C++, CUDA, PyTorch, JAX, TensorFlow, Keras, Quantization, Pruning, Model Compression, Ml Compilers
**Posted:** 2026-06-16

> Build and optimize ML infrastructure for autonomous vehicles, focusing on model optimization, compilers, and deployment of large models to Nuro's self-driving fleet. Requires 3+ years ML optimization experience and strong Python/C++/CUDA skills.

## Job Description

## About the Work

- Optimize Nuro’s autonomy stack with cutting-edge optimization techniques like quantization, distillation, and model compression.
- Work with autonomy engineers to optimize, validate, and deploy large language models.
- Develop and maintain a world-class model compiler framework, FTL.
- Write robust, high quality software to increase our confidence in our vehicle’s ability to navigate safely on-road.
- Collaborate closely with machine learning domain experts and engineers across behavior, perception and mapping to design and implement end-to-end learned ML solutions.

## About You

- 3+ years of relevant experience in ML optimization infrastructure.
- Experience with ML optimization techniques such as quantization and pruning, and ML compilers.
- Experience maintaining, profiling, and optimizing GPU ML compilers & runtimes.
- Proficient in Python and working experience with C++ and CUDA.
- Working experience deep learning frameworks (like PyTorch, Jax, Tensorflow, Keras).
- Proficient in Python and working experience with C++.
- You are passionate about accelerating the benefits of robotics for everyday life.

## Compensation and Benefits

- Base pay range: $193,930 - $352,290
- Annual performance bonus
- Equity
- Competitive benefits package

## Similar roles

- [Senior/Staff Machine Learning Engineer, Sensor Simulation](https://hotfix.jobs/jobs/67882c92-6808-4924-b827-81f676c9980f) - Nuro - Mountain View, CA - $194k – $291k/yr
- [Senior/Staff Software Engineer, ML Data](https://hotfix.jobs/jobs/d3f8ceb2-77cd-4ff3-9198-7dfe9ebe891e) - Nuro - Mountain View, CA - $194k – $352k/yr
- [Staff Software Engineer, AI-Core](https://hotfix.jobs/jobs/e2048905-d170-4047-a7cc-dec07104ee7a) - Okta - San Francisco, CA - $194k – $267k/yr
- [Staff Machine Learning Engineer, Credit Products (Square Financial Services)](https://hotfix.jobs/jobs/bd2bf98f-d359-4412-bc20-6d789f2f37cd) - Square - Seattle, WA - $195k – $343k/yr
- [Staff Software Engineer](https://hotfix.jobs/jobs/8ed0c488-5b98-4e7e-a7c1-058995ebc01f) - Databricks - New York, NY - $192k – $260k/yr

**Apply:** https://hotfix.jobs/jobs/be00f07f-f7b1-4685-9bf0-902f9d303a43
**Canonical:** https://hotfix.jobs/jobs/be00f07f-f7b1-4685-9bf0-902f9d303a43