Senior/Staff Software Engineer, ML Infrastructure, Optimization

• Optimize Nuro's autonomy stack with cutting-edge optimization techniques like quantization, distillation, and model compression.
• 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 confidence in the vehicle's ability to navigate safely on-road.

This role pushes the autonomous future forward by building core infrastructure for machine learning model life cycles, contributing to Nuro's mission of making autonomy accessible to all and empowering a safer, richer, and more connected future.

• 3+ years of relevant experience in ML optimization infrastructure.
• 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 with deep learning frameworks (like PyTorch, Jax, Tensorflow, Keras).
• Passionate about accelerating the benefits of robotics for everyday life.

Base pay range between $193,930 and $352,290, plus annual performance bonus, equity, and a competitive benefits package. Nuro celebrates differences and is committed to a diverse workplace that fosters inclusion and psychological safety for all employees.

Nuro is a self-driving technology company on a mission to make autonomy accessible to all, building the world's most scalable driver by combining cutting-edge AI with automotive-grade hardware to license its core technology for applications like robotaxis and commercial fleets.