Software Engineer, ML Infrastructure, Optimization
160k – 241kMountain View, CAML EngineeringOnsite2+ YOE
Summary
Build and optimize ML infrastructure for autonomous vehicles, focusing on model optimization, compilers, and deployment across the autonomy stack. Requires 2+ years in ML optimization and strong Python/C++/CUDA skills.
About the role
About the Work
- 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.
About You
- 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++.
- You are passionate about accelerating the benefits of robotics for everyday life.
Compensation and Benefits
- Base pay range: $160,360 - $240,540
- Annual performance bonus
- Equity
- Competitive benefits package
Skills
PythonC++CUDAPyTorchJAXTensorFlowKerasquantizationmodel pruningML compilers
Similar roles at this salary range
All ML Engineering jobs →Senior Machine Learning Operations Engineer
Build and operate Mercury's real-time ML inference platform for fraud risk decisioning. Own model deployment, observability, and lifecycle tooling with strong backend Python fundamentals.
167k – 208kSan Francisco, CA +2ML EngineeringHybrid5+ YOESQLSHAP
AI Engineer, Evaluation
Design and implement evaluation frameworks and pipelines for AI systems using Evaluation-Driven Development. Build Python-based test suites, LLM graders, and measurement systems that guide prompt iteration and production deployment decisions.
150k – 250kSan Francisco, CA +1ML EngineeringHybrid2+ YOEPythonAI Systems