Nuro

Senior/Staff Software Engineer, ML Infrastructure, Optimization

Nuro
Apply
3 months ago
Mountain View, CA, USASenior / Staff+
H1B sponsor

Base Salary

$194k - $352k/yr

Responsibilities

  • Optimize Nuro’s autonomy stack using techniques including quantization, distillation, and model compression.
  • Optimize, validate, and deploy large language models in collaboration with autonomy engineers.
  • Develop and maintain the FTL model compiler framework.
  • Write robust, high-quality software that improves confidence in vehicle on-road navigation.
  • Collaborate with machine learning experts and engineers across behavior, perception, and mapping to implement end-to-end learned ML solutions.

Requirements

  • At least 3 years of relevant experience in ML optimization infrastructure.
  • Experience with ML optimization techniques such as quantization and pruning, as well as 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, and Keras.

Benefits

  • Eligible for an annual performance bonus, equity, and a competitive benefits package.

Tech Stack

C++KerasPythonPyTorchTensorFlow
Nuro

About Nuro

1,001-5,000 employees

Nuro builds the Nuro Driver, a Level 4 self-driving platform for robotaxis, commercial fleets, and personally owned vehicles. It licenses its autonomy software and integrates it with automotive-grade hardware for automakers and mobility platforms. Founded in 2016 and headquartered in Mountain View, California, Nuro is privately held and has operated real-world self-driving deployments.

Contact me