Software Engineer, ML Infrastructure, Optimization

Highlights

• Build core ML infrastructure powering Nuro's universal autonomy platform.
• Deploy optimized models to Nuro's fleet of self-driving robots.
• Design end-to-end learned solutions with behavior, perception and mapping teams.

What they offer

• Annual performance bonus
• Equity
• Competitive benefits package

Ideal candidate

You are a collaborative ML infrastructure builder excited by full-stack model work from pipeline to on-road deployment. You care deeply about safe autonomy and accelerating robotics benefits for everyday life.

Requirements

• 2+ 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