Nuro

Software Engineer, ML Infrastructure, Optimization

Nuro
Apply
3 months ago
Mountain View, CA, USAMid Level
H1B sponsor

Base Salary

$160k - $241k/yr

Responsibilities

  • Optimize Nuro’s autonomy stack using quantization, low-precision inference, model pruning, and related techniques.
  • Collaborate with autonomy engineers to optimize, validate, and deploy large language models.
  • Develop and maintain the FTL model compiler framework.
  • Write robust, high-quality software that improves confidence in vehicle navigation safety.
  • Work with behavior, perception, and mapping experts to design and implement end-to-end learned machine learning solutions.

Requirements

  • At least 2 years of relevant experience in ML optimization infrastructure.
  • Experience with ML optimization techniques such as quantization and pruning, and with 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.
  • Passion for accelerating the benefits of robotics for everyday life.

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