Tech Lead Manager, ML Training Infrastructure
4 days ago
Who We Are
Nuro is a self-driving technology company on a mission to make autonomy accessible to all. Founded in 2016, Nuro is building the world's most scalable driver, combining cutting-edge AI with automotive-grade hardware. Nuro licenses its core technology, the Nuro Driver, to support a wide range of applications, from robotaxis and commercial fleets to personally owned vehicles. With technology proven over years of self-driving deployments, Nuro gives the automakers and mobility platforms a clear path to AVs at commercial scale—empowering a safer, richer, and more connected future.
About the Role
Nuro is seeking an experienced Technical Lead Manager with deep expertise in quantized training and model compression to join our ML Infrastructure team. In this role, you will drive the adoption of state-of-the-art quantization techniques, enabling training and deployment of highly-efficient models that power the Nuro Driver. You will lead technical strategy, mentor a team of engineers and researchers, and partner closely with research and product groups to ensure our ML infrastructure is optimized for both cutting-edge research and real-time deployment on autonomous vehicles.
About the Work
As TLM for Nuro's ML Training Infrastructure you will improve model training efficiency and drive the adoption of state-of-the-art quantization techniques. This will include:
- Setting technical direction for the Training Infrastructure team.
- Staying ahead of emerging research and evaluating new methods.
- Establishing telemetry to root-cause quality regressions in lower precision training.
- Driving the adoption of quantized training methods (e.g., AWQ, AQT, GPTQ) across Nuro's ML infrastructure to accelerate model training and inference.
- Leading the design and implementation of efficiency initiatives for model training, including low-bit quantization, pruning, and knowledge distillation, for both research and production workloads.
- Collaborating cross-functionally with research, infrastructure, and product teams balancing accuracy, latency, and resource constraints.
- Mentoring and growing a high-performing team of engineers and researchers.
About You
- 6+ years of professional or research experience in ML infrastructure, distributed training, or ML systems engineering.
- Hands-on experience with quantization methods, including Activation-Aware Weight Quantization (AWQ), Accurate Quantized Training (AQT), FP-8 training, or related methods.
- Knowledge of broader model compression techniques, such as structured/unstructured pruning and knowledge distillation.
- Experience building or maintaining quantization libraries (e.g., AQT, bitsandbytes, NVIDIA Transformer Engine, DeepSpeed Compression).
- Understanding of calibration and scaling strategies for quantized models to minimize accuracy loss.
Bonus Points
- Advanced degree (Ph.D. or strong M.Sc. with research experience) in Computer Science, Electrical Engineering, or related fields.
- Knowledge of sparse networks and complementary model compression techniques (e.g., AdaRound, BRECQ, structured pruning).
- Published work or open-source contributions in quantization methods (e.g., AWQ, AQT, GPTQ, SmoothQuant, ZeroQuant).
At Nuro, your base pay is one part of your total compensation package. For this position, the reasonably expected base pay range is between $235,030 and $352,290 for the level at which this job has been scoped. Your base pay will depend on several factors, including your experience, qualifications, education, location, and skills. In the event that you are considered for a different level, a higher or lower pay range would apply. This position is also eligible for an annual performance bonus, equity, and a competitive benefits package.
At Nuro, we celebrate differences and are committed to a diverse workplace that fosters inclusion and psychological safety for all employees. Nuro is proud to be an equal opportunity employer and expressly prohibits any form of workplace discrimination based on race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, genetic information, disability, veteran status, or any other legally protected characteristics. #LI-DNP
-
Mountain View, United States Nuro Full timeWho We Are Nuro is a self-driving technology company on a mission to make autonomy accessible to all. Founded in 2016, Nuro is building the world’s most scalable driver, combining cutting-edge AI with automotive-grade hardware. Nuro licenses its core technology, the Nuro Driver™, to support a wide range of applications, from robotaxis and commercial...
-
Tech Lead Manager, ML Training Infrastructure
2 weeks ago
Mountain View, CA, United States Nuro Full timeWho We Are Nuro is a self-driving technology company on a mission to make autonomy accessible to all. Founded in 2016, Nuro is building the world's most scalable driver, combining cutting-edge AI with automotive-grade hardware. Nuro licenses its core technology, the Nuro Driver™, to support a wide range of applications, from robotaxis and commercial...
-
Mountain View, United States Google Full timeAndroid Software Engineering Tech Lead Manager, AI/ML Responsibilities Grow the local team expertise and projects, mentoring the team and removing bottlenecks. Improve the team's design, code, and engineering practices. Interface with cross-functional and remote teams, engineering managers, and stakeholders to influence cross-team priority and roadmap....
-
Engineering Manager
2 weeks ago
Mountain View, United States Moveworks Full timeA leading AI company in Mountain View, CA, is hiring an Engineering Manager for Machine Learning Infrastructure. You will lead the development of a state-of-the-art platform powering conversational AI, ensuring scalability and performance of ML systems. The ideal candidate will have a Master's or Ph.D. in related fields, along with 5+ years of industry...
-
Lead Machine Learning Infrastructure Engineer
2 weeks ago
Mountain View, United States Convene Full timeAbout Convene Inc. Convene, Inc. is a Tampa based, award-winning technology services organization with offices and resources throughout the US, Mexico, and India. We have successful, referenceable customers, competitive benefits, and high-growth opportunities. Lead ML Infrastructure Engineer Mountain View, CA/ Dallas, TX / Chicago, IL / NYC, NY About the...
-
Mountain View, United States Google Inc. Full timeSenior Staff Software Engineer, ML Data Quality Infrastructure corporate_fare Google place Mountain View, CA, USA Apply Bachelor's degree or equivalent practical experience. 8 years of experience in software development. 7 years of experience leading technical project strategy, ML design, and working with ML infrastructure (e.g., model deployment, model...
-
Mountain View, California, United States Google Full time $248,000 - $349,000 per yearMinimum qualifications:Bachelor's degree or equivalent practical experience.8 years of experience in software development.7 years of experience leading technical project strategy, ML design, and working with ML infrastructure (e.g., model deployment, model evaluation, data processing, debugging, fine tuning).5 years of experience with one or more of the...
-
Mountain View, United States Nuro Full timeSoftware Engineer, ML Infrastructure, Optimization Who We AreNuro is a self‑driving technology company on a mission to make autonomy accessible to all. Founded in 2016, Nuro is building the world’s most scalable driver, combining cutting‑edge AI with automotive‑grade hardware. Nuro licenses its core technology, the Nuro Driver™, to support a wide...
-
Senior Software Engineer, ML Evaluation
1 week ago
Mountain View, United States Waymo Full timeWaymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver-The World's Most Experienced Driver™-to improve access to mobility while saving thousands of lives now lost to traffic crashes. The Waymo...
-
Senior Software Engineer, ML Evaluation
2 weeks ago
Mountain View, United States Waymo Full timeWaymo is an autonomous driving technology company with the mission to be the worlds most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo DriverThe Worlds Most Experienced Driverto improve access to mobility while saving thousands of lives now lost to traffic crashes. The Waymo Driver...