Software Engineer (ML Infrastructure, Optimization)
Posted Updated
- 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, low precision inference, and model pruning
- 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
- Working experience deep learning frameworks (like PyTorch, Jax, Tensorflow, Keras)
- 2+ years of relevant experience in ML optimization infrastructure
- Experience with ML optimization techniques such as quantization and pruning, and ML compilers
- Proficient in Python and working experience with C++ and CUDA
- Proficient in Python and working experience with C++
- Experience maintaining, profiling, and optimizing GPU ML compilers & runtimes
- You are passionate about accelerating the benefits of robotics for everyday life