Senior/Staff Software Engineer, ML Infrastructure, Optimization

Highlights

• Optimize Nuro's autonomy stack with cutting-edge model compression techniques.
• Deploy optimized models on Nuro's fleet of self-driving robots.
• Build end-to-end learned ML solutions across behavior, perception and mapping.

What they offer

• Annual performance bonus
• Equity
• Competitive benefits package

Ideal candidate

Experienced ML infrastructure engineer passionate about accelerating robotics for everyday life. Collaborative builder who thrives taking models from pipeline to safe on-road deployment.

Requirements

• 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