Senior/Staff Software Engineer, ML Infrastructure, Optimization
Mountain View, CA$193.9k - $352.3k full-timePosted 4mo ago
About the role
Senior/Staff Software Engineer focused on ML infrastructure and optimization for Nuro's autonomy stack. Builds and optimizes core ML infrastructure, model pipelines, and deployment of optimized models for Nuro’s fleet, collaborating with behavior, perception, and mapping teams.
Responsibilities
- Optimize Nuro’s autonomy stack with cutting-edge optimization techniques like quantization, distillation, and model compression.
- 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.
Qualifications
- 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 deep learning frameworks (like PyTorch, Jax, Tensorflow, Keras).
- Proficient in Python and working experience with C++.
Skills
Required:ML optimization infrastructureQuantizationPruningML compilers (e.g., MLIR, in-house compilers)PythonC++CUDAPyTorch
Nice to have:JaxTensorFlow
Benefits
- Base pay range: $193,930 - $352,290 (base pay varies by level, experience, location, etc.).
- Eligible for annual performance bonus, equity, and a competitive benefits package.
Last updated Jul 8, 2026. Interested? Apply on the company site.
About Nuro
Autonomy for all. All roads, all rides.
Driving the future of autonomy with the most scalable AI driver.
AutonomyAI driverDelivery robotsRobotaxiAutonomous vehicles
More open roles at Nuro
- Senior Software Engineer, Autonomy Behavior Testing4d agoMountain View, CA · Systems
- Staff Robotaxi Program Manager4d agoMountain View, CA · Autonomy
- Staff Autonomy Program Manager4d agoMountain View, CA · Autonomy
- Senior Embedded Software Engineer1w agoMountain View, CA · Controls, Pose and ESWE
- Senior Radar Systems Engineer1w agoMountain View, CA · Sensors Engineering