Drimva

Nuro

Software Engineer, ML Infrastructure, Optimization

$160K–$241K

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

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

About the job

Build and optimize machine-learning infrastructure and models for Nuro’s autonomous driving stack, including deploying optimized models on its self-driving fleet.

What you'll do

  • Optimize the autonomy stack using techniques such as quantization, low-precision inference, and model pruning; validate and deploy large language models.
  • Develop and maintain the FTL model compiler framework and optimize GPU ML compilers and runtimes.
  • Build robust ML lifecycle infrastructure and collaborate across autonomy teams on end-to-end learned solutions.

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.
  • Proficient in Python; working experience with C++ and CUDA.
  • Working experience with deep learning frameworks such as PyTorch, Jax, Tensorflow, or Keras.

Pay

$160,360 and $240,540 (USD, base salary)

Tools

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

AI focus

Optimize, validate, and deploy machine-learning models, including large language models, for autonomous driving.

Model-written summary · the original posting is authoritative

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