Drimva

Nuro

Senior/Staff Software Engineer, ML Infrastructure, Optimization

$194K–$352K

Mountain View, California (HQ) +1 · Staff · Full-time · Posted Jun 16, 2026

Location
Mountain View, California (HQ)
Nuro HQ - Mountain View, CA
Experience
3+ years of relevant experience in ML optimization infrastructure.

About the job

Build and optimize machine-learning infrastructure, compilers, and model deployment workflows for Nuro’s autonomous driving systems.

What you'll do

  • Optimize the autonomy stack using techniques such as quantization, distillation, and model compression; validate and deploy large language models.
  • Develop and maintain the FTL model compiler framework and optimize GPU ML compilers and runtimes.
  • Collaborate across autonomy teams to build and deploy end-to-end ML solutions and robust software for safe on-road navigation.

What they ask for

  • Experience with ML optimization techniques such as quantization and pruning, and ML compilers.
  • Experience maintaining, profiling, and optimizing GPU ML compilers and runtimes.
  • Proficiency in Python and working experience with C++ and CUDA.
  • Working experience with deep learning frameworks such as PyTorch, Jax, Tensorflow, or Keras.

Pay

USD $193,930 and $352,290 base salary

Tools

FTL, Python, C++, CUDA, PyTorch, Jax, Tensorflow, Keras

AI focus

Optimize, validate, and deploy ML models, including large language models, for autonomous driving; develop ML infrastructure and compilers.

Model-written summary · the original posting is authoritative

First seen Oct 10, 2026 · Last verified Nov 5, 2026 (UTC)