Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Technical Lead Manager, ML Training Infrastructure

$223k - $334k

Nuro

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 $222,775 and $333,925 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

Vacancy posted more than 2 months ago
Similar jobs that could be interesting for youBased on the Technical Lead Manager, ML Training Infrastructure in Mountain View, CA vacancy
  • $238k - $302k

     ...goals. We collaborate across teams to bring ML to production systems and build what is...  ...machine learning models to deliver training and evaluation data for the Waymo driver...  ...autonomous vehicle's behavior. Develop ML infrastructure to support performant models.... 
    Training
    Full time
    Work experience placement
    Remote work

    Waymo

    Mountain View, CA
    2 days ago
  • $237k - $329k

    Robotics Automation Technical Lead Manager, Platforms Infrastructure Google Sunnyvale, CA, USA Qualifications Bachelor’s degree or equivalent practical experience...  ...skills, experience, and relevant education or training. Your recruiter can share more about the specific... 
    Training
    Full time
    Contract work
    Remote work
    Worldwide
    Flexible hours

    Google Inc.

    Sunnyvale, CA
    4 days ago
  • $156k - $229k

    Technical Program Manager III, Networking Infrastructure, Google Cloud Google, Sunnyvale, CA, USA Mid Experience...  .... Experience in leading and successfully...  ...Google products like AI/ML, Cloud, Search, Ads, Maps...  ...and relevant education or training. Your recruiter can share... 
    Training
    Full time

    Google Inc.

    Sunnyvale, CA
    1 day ago
  • $163k - $237k

     ...Technical Program Manager III, GPU Infrastructure Reliability Mid Experience driving progress, solving...  ...software engineering and ML infrastructure projects....  ...technical expertise to lead complex, multi-...  ...and relevant education or training. US: $163000 - $237000 (USD... 
    Training
    Temporary work

    Google

    Sunnyvale, CA
    4 days ago
  • $138k - $198k

     ...Technical Program Manager II, Infrastructure and Capacity, Platforms and Devices Mid Experience...  ...within Machine Learning (ML) infrastructure, specifically...  ...Manager at Google, you lead complex, multi-disciplinary...  ...and relevant education or training. US: $138000 - $198000 (... 
    Training
    Immediate start

    Google

    Mountain View, CA
    4 days ago
  •  ...the Role We are seeking a Staff Technical Program Manager (TPM) to leadAVMLInfrastructureprograms...  ...and execution for large-scale ML infrastructure — including training pipelines, model lifecycle...  ...Responsibilities Program Leadership Lead end-to-end strategic planning and... 
    Training
    Relocation
    Relocation package

    General Motors

    Sunnyvale, CA
    2 days ago
  • $240k - $334k

    Technical Lead Manager, Machine Learning, Memory Subsystem Design Sunnyvale, CA, USA Qualifications...  ...party IP providers. The AI and infrastructure team is redefining what’s possible. We...  ...experience, and relevant education or training. Your recruiter can share more about... 
    Training
    Full time
    Worldwide

    Google Inc.

    Sunnyvale, CA
    2 days ago
  • $298k - $368k

     ...think deeply and solve complex technical challenges in areas like...  ...and feed it to the Perception ML models. We specialize in low‑...  ...stack. We are looking for a Tech‑Lead Manager with a CPU, GPU, or system‑...  ...location, experience, relevant training and education, and skill... 
    Training
    Full time
    Remote work

    Waymo

    Mountain View, CA
    4 days ago
  • $238k - $302k

     ...Senior Staff TLM. You will: Lead and build a team of ML engineers in charge of...  ...automatically generate suitable training data for rare instance...  ...of experience leading technical teams and setting technical...  ...directions 2+ years of people management experience We prefer:... 
    Training
    Full time
    Work at office
    Immediate start
    Remote work

    Waymo

    Mountain View, CA
    4 days ago
  • $250.75k - $318.6k

     ...learning-based explorations. As a Technical Lead, you'll have an impact on...  ...report to an Engineering Manager. You will Working closely with...  ...workflows and building metrics for ML models Salary Range The...  ..., experience, relevant training and education, and skill level... 
    Training
    Full time
    Remote work

    Waymo

    Mountain View, CA
    1 day ago
  • $235.03k - $352.29k

     ...and deploy algorithms to measure the performance of the ML models that power our self‑driving car’s behavior,...  ...machine learning. Work cross‑functionally with Autonomy and Infrastructure engineers to set the technical roadmap that will continuously improve and unify the... 

    Icehouseventures

    Mountain View, CA
    15 hours ago
  • $235.03k - $352.29k

    Nuro is looking for a Technical Lead Manager in Mountain View, California. You will lead the ML Infrastructure team, driving strategy and optimizing workflows for autonomous driving technology. Candidates should have 6+ years in ML infrastructure, backend engineering,... 

    I did my part and supported the Regular Toilet

    Mountain View, CA
    4 days ago
  • $251k - $310k

     ...internationally. In this hybrid role you will report to a Sr Staff Technical Lead Manager. You will: Own a specific domain within scene...  ...factors, including exact work location, experience, relevant training and education, and skill level. Waymo employees are also eligible... 
    Training
    Full time
    Temporary work

    Waymo

    Mountain View, CA
    15 hours ago
  • $300k - $334k

    Senior Technical Program Manager Lead, Gemini Audio, DeepMind Location preference: Mountain View, CA,...  ...audio space. Experience in model training, model releases, or data science....  ...of the programs from interface to infrastructure, serving, and customer issues, and... 
    Training

    Google Inc.

    Mountain View, CA
    4 days ago
  • $238k - $302k

     ...destinations. We think deeply and solve complex technical challenges in areas like robotics,...  ...role you will report to a Technical Lead Manager, Senior Staff Engineer. You Will...  ...exact work location, experience, relevant training and education, and skill level. Your recruiter... 
    Training
    Full time
    Remote work

    Waymo

    Mountain View, CA
    1 day ago
  • $231.9k - $298.1k

     ...Sunnyvale, California is hiring a Technical Lead Manager to oversee the perception...  ...vehicle software infrastructure. This role demands a candidate...  ...with a strong background in ML and management, keen on driving...  ...perception model, leading training efforts, and working... 
    Training

    Applied Intuition

    Sunnyvale, CA
    4 days ago
  • $240k - $334k

    Technical Lead, Custom Digital Circuit Design, Mixed Signal corporate_fare...  ...to shape the future of AI/ML hardware acceleration. You...  ...circuit design. The AI and Infrastructure team is redefining what’s possible...  ..., and relevant education or training. Your recruiter can share... 
    Training
    Full time
    Worldwide

    Google Inc.

    Sunnyvale, CA
    4 days ago
  • $147k - $211k

     ..., Video and Display Modeling Infrastructure Google - Mountain View, CA, USA...  ...Computer Science or related technical fields. 2 years of...  ...collection, backfilling, model training, feature iteration, inference...  ...modeling. Assist clients (AViD ML teams) with onboarding the latest... 
    Training

    Google Inc.

    Mountain View, CA
    1 day ago
  • $141k - $202k

    Software Engineer III, AI/ML, AI and Infrastructure corporate_fare Google place...  ...Computer Science or related technical fields. 2 years of experience...  ...the future of world-leading hyperscale computing, with...  ...and relevant education or training. Your recruiter can share more... 
    Training
    Full time
    Worldwide

    Google Inc.

    Sunnyvale, CA
    1 day ago
  • $230k - $292k

     ...Staff Technical Program Manager, Simulation Infrastructure Resource Management Waymo is an autonomous driving technology company with the mission to be the...  ...including exact work location, experience, relevant training and education, and skill level. Your recruiter can... 
    Training
    Full time
    Temporary work
    Remote work

    Waymo

    Mountain View, CA
    4 days ago
  • $163k - $237k

     ...Bachelor's degree in a technical field or...  ...experience in program management. Experience in...  ...intelligence. Experience leading teams through the...  ...cutting-edge AI/ML modeling, or AI...  ...and expand our infrastructure, while...  ...relevant education or training. US: $163,000 -... 
    Training

    Google

    Mountain View, CA
    1 day ago
  • $163k - $237k

     ...Bachelor's degree in a technical field, or...  ...experience in program management. Experience with AI/ML. Experience with resource...  ...expertise to lead complex, multi‑disciplinary...  ...education or training. US: $163,000 - $23...  ...Partner with modeling, infrastructure, and research teams... 
    Training

    Google

    Mountain View, CA
    4 days ago
  • $192k - $278k

     ...Bachelor's degree in a technical field, or...  ...experience in program management. Preferred Qualifications...  ...expertise to lead complex, multi-...  ...engineers. The AI and Infrastructure team is redefining...  ...education or training. Your recruiter can...  ..., contribute to ML system roadmap by... 
    Training
    Full time
    Worldwide

    Google

    Sunnyvale, CA
    3 days ago
  • As a Technical Program Manager at RadixArk, you'll drive the execution of complex...  ...across our inference and training infrastructure. You'll partner closely...  ...shoulder with engineering leads on technical trade‑off decisions...  ...systems, or AI/ML infrastructure; able to read... 
    Training
    Full time

    RadixArk

    Palo Alto, CA
    2 days ago
  • $192k - $279k

     ...Bachelor's degree in a technical field, or...  ...experience in program management. 8 years of experience...  ...for cloud infrastructure. About The Job A...  ...technical expertise to lead complex, multi‑...  ...relevant education or training. US: $192,000 - $...  ...related to ML Fleet capacity management... 
    Training
    Temporary work

    Google

    Sunnyvale, CA
    4 days ago
  • $240k - $334k

    Technical Program Manager, Search Velocity and AI Acceleration Google...  ...managing AI/ML or Large Language Model...  ...Manager at Google, you lead complex, multi-disciplinary...  ...and expand our infrastructure, while maintaining a...  ...relevant education or training. US: $240000 - $3340... 
    Training
    Shift work

    Google

    Mountain View, CA
    2 days ago
  • $163k - $237k

    Technical Program Manager III, AI/ML, Google Ads corporate_fare Google place Mountain View...  ...technical expertise to lead complex, multi-...  ...and relevant education or training. US: $163000 - $237000 (...  ...Partner with modeling, infrastructure, and research teams to drive... 
    Training

    Google Inc.

    Mountain View, CA
    3 days ago
  • $192k - $278k

     ...Bachelor's degree in a technical field, or...  ...experience in program management. Preferred qualifications...  ...expertise to lead complex, multi-...  ...engineers. The AI and Infrastructure team is redefining...  ...education or training. Your recruiter can...  ..., contribute to ML system roadmap by... 
    Training
    Full time
    Worldwide

    Google Inc.

    Sunnyvale, CA
    4 days ago
  •  ..., we are working on training and scaling up multimodal...  ...We're looking for a Technical Program Manager to partner closely...  ...across research, infrastructure, and product engineering...  ...scientists, ML engineers, and infrastructure...  ...Us Work alongside leading researchers pushing... 
    Training

    Luma AI

    Palo Alto, CA
    1 day ago
  • $174k - $253k

    Senior Software Engineer, AI/ML Training Infrastructure Google Mountain View, CA, USA Apply Bachelor’s degree or equivalent practical experience...  ...: Master's degree or PhD in Computer Science or related technical field. 5 years of experience with data structures and... 
    Training

    Google Inc.

    Mountain View, CA
    1 day ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Technical Lead Manager, ML Training Infrastructure. Be the first to apply!