Senior/Staff Software Engineer, ML Infrastructure, Optimization
Core
Design, build, and deploy core infrastructure components for the machine learning model life cycle, focusing on optimizing autonomy stacks and deploying models on self-driving robots.
Role type
Senior/Staff Software Engineer, ML Infrastructure, Optimization
Builds
Robust and scalable model pipelines, optimized ML models, and a world-class model compiler framework (FTL) for Nuro's fleet of self-driving robots.
Domain
Autonomous driving, robotics, machine learning infrastructure
Deliverable
production ML models
Required skills
ML optimization techniques (quantization, pruning, model compression), ML compilers, GPU ML compilers & runtimes, Python, C++, CUDA, deep learning frameworks (PyTorch, Jax, Tensorflow, Keras)
Preferred skills
Experience with large language models, end-to-end learned ML solutions
Technologies
FTL, PyTorch, Jax, Tensorflow, Keras, CUDA
Responsibilities
Optimize Nuro's autonomy stack with cutting-edge optimization techniques; work with autonomy engineers to optimize, validate, and deploy large language models; develop and maintain the FTL model compiler framework; write robust software to increase confidence in vehicle navigation; collaborate with ML domain experts to design and implement end-to-end learned ML solutions.
Seniority
Senior/Staff, hands-on IC
