Software Engineer (ML Infrastructure, Optimization)

Join Nuro, a company dedicated to revolutionizing transportation through autonomous delivery. As a Software Engineer on the Autonomy ML Infrastructure team, you will design, build, and deploy core infrastructure components for the machine learning model life cycle. You will work across the full stack of machine learning solutions, optimize Nuro's autonomy stack, and collaborate with machine learning domain experts. Enjoy benefits such as a free Caltrain pass, company stock options, and health insurance.

• Collaborate with cross-functional teams to design, build, and deploy core infrastructure components in the machine learning model life cycle.
• Optimize Nuro’s autonomy stack using cutting-edge optimization techniques, and work with autonomy engineers to optimize, validate, and deploy large language models.
• Develop and maintain a world-class model compiler framework, FTL, and write robust, high-quality software to increase confidence in vehicle navigation.
• You are passionate about accelerating the benefits of robotics for everyday life
• 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++
• Proficient in Python and working experience with C++ and CUDA
• 2+ years of relevant experience in ML optimization infrastructure
• Working experience deep learning frameworks (like PyTorch, Jax, Tensorflow, Keras)