Senior/Staff Software Engineer (ML Infrastructure, Optimization)

• The Autonomy ML Infrastructure team is responsible for building & improving the core infrastructure for autonomy teams at Nuro
• In this role, you will work closely with teams across Nuro, to design, build and deploy core infrastructure components in machine learning model life cycle, to push the autonomous future forward
• You will have an opportunity to work across the full stack of machine learning solutions - from designing robust and scalable model pipelines to building to deploying the optimized models on Nuro’s fleet of self-driving robots!
• 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 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

Benefits

• Free Caltrain pass and commuter benefits
• Company stock options
• Work from home opportunities
• Health insurance- Proficient in Python and working experience with C++ and CUDA
• Working experience deep learning frameworks (like PyTorch, Jax, Tensorflow, Keras)
• Proficient in Python and working experience with C++
• You are passionate about accelerating the benefits of robotics for everyday life
• Experience maintaining, profiling, and optimizing GPU ML compilers & runtimes
• Experience with ML optimization techniques such as quantization and pruning, and ML compilers
• 3+ years of relevant experience in ML optimization infrastructure