# Software Engineer, ML Infrastructure, Optimization

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

> Build and optimize ML infrastructure for autonomous vehicles, focusing on model optimization, compilers, and deployment across the autonomy stack. Requires 2+ years in ML optimization and strong Python/C++/CUDA skills.

## Job Description

## About the Work

- Optimize Nuro’s autonomy stack with cutting-edge optimization techniques like quantization, low precision inference, and model pruning.
- 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

- 2+ 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: $160,360 - $240,540
- Annual performance bonus
- Equity
- Competitive benefits package

## Similar roles

- [Software Engineer, Model Performance Tooling](https://hotfix.jobs/jobs/26ede8a1-a8ce-494b-bcb9-87270ba2d333) - Baseten - San Francisco, CA - $160k – $200k/yr
- [Software Engineer, Agents](https://hotfix.jobs/jobs/c48c3d5d-396e-4051-938f-a677c38c294d) - Harvey - San Francisco, CA - $161k – $242k/yr
- [Machine Learning Engineer, Monetization Engineering](https://hotfix.jobs/jobs/717f63ae-bbd1-483c-9c49-b2a025ed411d) - Pinterest - San Francisco, CA - $163k – $286k/yr
- [Software Engineer](https://hotfix.jobs/jobs/823e27ee-a56e-4d3e-ac4d-dacbce6de23d) - Abridge - San Francisco, CA - $157k – $184k/yr
- [Machine Learning Engineer, Relevance and Personalization](https://hotfix.jobs/jobs/e55bff4d-9948-4fd5-8f5e-51f0f5c17db4) - Airbnb - Remote - $166k – $195k/yr

**Apply:** https://hotfix.jobs/jobs/a6b5f42b-e6f7-497c-adda-7a35b24c3eb4
**Canonical:** https://hotfix.jobs/jobs/a6b5f42b-e6f7-497c-adda-7a35b24c3eb4