Software Engineer, ML Infrastructure, Optimization
Mountain View, CA$160.4k - $240.5kfull-timePosted 4mo ago
About the role
Role focused on building and improving core ML infrastructure for autonomy pipelines, enabling design, build, and deployment of optimized models for Nuro's autonomous fleet.
Responsibilities
- 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.
Qualifications
- 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 deep learning frameworks (like PyTorch, Jax, Tensorflow, Keras).
- Proficient in Python and working experience with C++.
Skills
Required:ML optimization infrastructureQuantizationPruningML compilersGPU profiling/optimizationPythonC++CUDA
Benefits
- annual performance bonus
- equity
- 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