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
  • $251k - $310k

     ...exciting opportunity for a Staff Technical Lead Manager to lead our ML Evaluation team. In this role, you...  ...evaluation platforms, scaling the critical infrastructure and metrics required, and partner...  ...location, experience, relevant training and education, and skill level.... 
    Training
    Full time
    Remote work

    Waymo

    Mountain View, CA
    1 day ago
  • $221k - $276k

     ...Opportunity The Simulation Infrastructure team is responsible...  .... We solve complex technical challenges to build services...  ...Key Responsibilities Lead an infrastructure team...  ...technical solutions. Manage a medium‑sized team (5...  ...experience, relevant training, and skill level.... 
    Training
    Full time
    Work at office
    Local area
    Remote work
    Flexible hours

    Waymo

    Mountain View, CA
    1 day ago
  • $251k - $310k

     ...simulation across 15+ U.S. states. As the Technical Lead Manager for Rider Growth, you will build, lead,...  ..., habitual riding. Collaborate with ML scientists and economists to design and...  ...work location, experience, relevant training and education, and skill level. Your recruiter... 
    Training
    Full time
    Remote work

    Waymo

    Mountain View, CA
    1 day ago
  • $281k - $356k

     ...World-class data analysis infrastructure is the bedrock of...  ...looking for an exceptional technical leader to drive the...  ...expansion. You will: Lead the architectural...  ...Simulation, Onboard, ML, and Commercialization...  ...experience, relevant training and education, and skill... 
    Training
    Full time
    Remote work

    Waymo

    Mountain View, 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
    3 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

    Sunnyvale, CA
    1 day ago
  • $193.93k - $291.15k

     ...expertise in large‑scale infrastructure, workload...  ...processing to join our ML Infrastructure team. In...  ...responsible for executing the technical strategy for automated...  ...and efficient feature management to accelerate the Nuro...  ...massive‑scale distributed training. Data & ETL: Designing... 
    Training

    Nuro

    Mountain View, CA
    5 days ago
  • $132.1k - $165.1k

     ...(heavily reliant on ML), Planning & Controls, Infrastructure & Simulation (critical for ML model training/validation), Systems...  ...product. Responsibilities Lead programs consisting...  ...diving into the technical details to drive...  ...resource allocation. Manage ambiguity by presenting... 
    Training
    Full time
    Contract work
    Local area

    Rivian

    Palo Alto, CA
    5 days ago
  • $200k - $249k

     ...and scalable batch and stream data processing infrastructure and solutions to support day to day ML operations including training, serving, evaluation and experimental systems...  ...needs, influence data team’s roadmap, and lead as well as execute on various projects. Build... 
    Training

    Moveworks.ai

    Mountain View, CA
    5 days ago
  •  ...Engineer, AI Compute Infrastructure About HeyGen At...  ...models—from multimodal training data pipelines to...  ...for launching and managing massive, heterogeneous...  ...with core ML frameworks such as...  ...Science or a related technical field. Demonstrated Tech Lead experience, driving... 
    Training
    Full time

    HeyGen

    Palo Alto, CA
    1 day ago
  •  ...hardware and robot systems to the infrastructure and state-of-the-art...  ...infrastructure that collects training data, keeps our robots running...  ...practices around security, cost management, and scalability Participate...  ...building infrastructure for ML workloads — GPU cluster management... 
    Training

    Rhoda AI

    Mountain View, CA
    5 days ago
  •  ...Software Engineer, Infrastructure Software for AI (...  ...Engineering, or a related technical field. 5+ years...  ...in technical lead roles, owning high...  ...AI workloads (training, fine‑tuning, and...  ...and GPU resource management. Drive architectural...  ...knowledge of major AI/ML frameworks (... 
    Training

    Intelliswift - An LTTS Company

    Sunnyvale, CA
    1 day 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
    4 days ago
  •  ...Platform and builds the shared infrastructure that helps DoorDash, Wolt,...  ...inference engines, fine‑tuning and training pipelines, GPU autoscaling...  ...fine‑tuning in a fast‑moving technical area where product needs,...  ...excellence. Partner closely with ML engineers, product engineers,... 
    Training
    Hourly pay
    Work at office
    Local area
    Remote work
    Flexible hours

    DoorDash

    Sunnyvale, CA
    1 day ago
  •  ...Google Cloud is leading this shift with Gemini...  ...it happen. As a Technical Delivery Lead, you...  ...AI engagements and manage cross‑stream...  ...AutoML, custom model training, and MLOps deployment...  ...‑grade AI/ML system experience....  ...Customer Engagement, Infrastructure Foundations, Cybersecurity... 
    Training
    Work experience placement
    Local area
    Shift work

    Accenture

    Mountain View, CA
    3 days ago
  •  ...Fidelity, T. Rowe Price, and other leading investors About the Role Evaluation Infrastructure plays a critical role at Nuro,...  ...stack spans both heuristic and ML-based approaches, covering everything...  ...broader AI Platform. As the Technical Lead, you will lead the team... 
    Temporary work
    Work experience placement
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    24 days ago
  • $160.36k - $240.54k

     ...agentic systems. Our team builds the AI infrastructure that enables autonomous coding agents,...  ...to debugging, knowledge discovery, and ML model improvement. We sit at the intersection...  ...‑horizon reasoning Integrate with ML training, evaluation, and data pipelines Create... 
    Training

    Nuro

    Mountain View, CA
    1 day ago
  • $192k - $278k

    Senior Technical Program Manager, AI2, Data Centers Experience owning...  ...expertise to lead complex, multi-disciplinary...  ...engineers. The AI and Infrastructure team is redefining...  ...relevant education or training. Your recruiter can share...  ...to Google's ML delivery roadmap by driving... 
    Training
    Full time
    Worldwide

    Google

    Sunnyvale, CA
    1 day ago
  • Google is seeking a Technical Manager for TPU DRAM to lead a team developing advanced DRAM subsystems. You will drive innovation in custom silicon development for machine learning products, collaborating with cross-functional teams and interfacing with third-party IP and... 

    Google

    Sunnyvale, CA
    1 day ago
  • $235.03k - $352.29k

     ...Rowe Price, and other leading investors. About the...  ...Autonomy Leader to drive the technical roadmap for the...  ...cognitive performance of the ML models powering our...  ...with Autonomy and Infrastructure engineers to set a roadmap...  ...execution. People Management: Experience in hiring... 
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    10 days ago
  • $213k - $263k

     ...reports to an Engineering Manager. Responsibilities...  ...generate high-quality ML and evaluation datasets...  ..., and slicing for training and testing the Waymo...  ...implement robust tools and infrastructure for data mining,...  ...Engineering, or a related technical field. Experience: 3+... 
    Training
    Full time

    Waymo

    Mountain View, CA
    1 day ago
  •  ...build the simulation infrastructure for physical AI. We develop...  ...unlimited, diverse training data and realistic...  ...built core systems at leading AI labs. We believe in...  ...You’ll make critical technical decisions, mentor engineers...  ...engineers Drive ML solution architecture... 
    Training

    FortyFive

    Mountain View, CA
    4 days ago
  •  ...allows Cerebras to deliver industry-leading training and inference speeds and...  ...to effortlessly run large-scale ML applications, without the hassle of managing hundreds of GPUs or TPUs. Cerebras...  .... The Role As a Senior Hardware Technical Program Manager at Cerebras, you... 
    Training

    CEREBRAS SYSTEMS INC.

    Sunnyvale, CA
    4 days ago
  • $235.03k - $352.29k

     ...Price, and other leading investors About...  ...technology. In an ML-first system, the...  ...diversity of its training and evaluation data...  ..., supporting infrastructure, and a suite of data...  ...in defining technical visions, creating...  ...monitoring and incident management Experience... 
    Training
    Immediate start
    Flexible hours

    Nuro

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

     .... The Architecture & ML Integration (AMI) team...  ...tackles critical shared infrastructure components You will:...  ...large scale products Lead the software...  ...subsystem between model training data preparation and model...  ...engineering projects as a technical lead. Deep... 
    Training
    Full time
    Remote work
    Flexible hours

    Waymo

    Mountain View, CA
    1 day ago
  • $190.8k - $267.1k

     ...of-the-art Machine Learning infrastructure. The Ads ML Platform team is at the...  ...: As the Senior Product Manager for the Ads ML Platform, you...  ...Scientists, enabling them to train, deploy, and iterate on models...  ...with prior focus internal technical products, developer tools,... 
    Training
    For contractors
    Work experience placement
    Work at office
    Remote work
    Home office
    Flexible hours

    GrabJobs

    San Jose, CA
    1 day ago
  • $190k - $260k

     ...only as good as the speed at which we can train it. Every improvement to our models –...  ...to large-scale world models – depends on infrastructure that turns thousands of hours of multimodal...  ...ready, streamable formats Partner with ML teams to scale new architectures from prototype... 
    Training
    Temporary work
    Work at office
    Visa sponsorship
    Flexible hours

    Kodiak

    Mountain View, CA
    12 days ago
  •  ...Radiology Partners, Cognita’s models are trained and validated on one of the world’s...  .... About the Role As a Member of Technical Staff on the ML Infrastructure team, you will build and operate...  ...at scale. Own systems that manage large-scale compute, storage, and data... 
    Training

    Cognita Imaging Inc.

    Palo Alto, CA
    5 days ago
  •  ...performing team that delivers infrastructure and performance...  ...companies.  As a Lead Infrastructure Engineer...  ...software, applications, and technical processes within the...  ...with integrated management, monitoring, and analytics...  ...and skills  Formal training or certification on infrastructure... 
    Training

    J.P. Morgan

    Palo Alto, CA
    10 days ago
  • $163k - $237k

    Technical Program Manager III, Machine Learning, Search corporate_fare Google...  ...your technical expertise to lead complex, multi-...  ...challenges and expand our infrastructure, while maintaining a universally...  ...and relevant education or training. Job Responsibilities Collaborate... 
    Training

    Google Inc.

    Mountain View, CA
    3 days 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!