Software Engineer, ML Infrastructure, OptimizationActive$160K–$241K

The opportunity

Nuro believes self-driving vehicles are the most immediate and profound opportunity for AI to drive positive change in the physical world. Safer streets, more time for what matters, and easier access to the world around us, that’s why we’re building a universal autonomy…

What you'll do

  • 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.

  • + years of relevant experience in ML optimization infrastructure.

What they're looking for

  • 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).