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
  • $235.03k - $352.29k

     ...Technical Lead Manager, ML Platform Infrastructure Mountain View, California (HQ) Nuro is a self-driving technology company on a mission to make autonomy...  ...wait times, and handle massive-scale distributed training. Data Dumping & ETL: Designing robust pipelines... 
    Training

    Nuro

    Mountain View, CA
    2 days ago
  •  ...re looking for a Tech Lead Manager (TLM) to own and drive...  ...your time on hands-on technical work and 30% on people...  ...models, agents, and ML systems that power...  ...driven planning workflows trained on each customer's...  ...training and inference infrastructure, set standards for... 
    Training
    Remote work
    Flexible hours

    TEEMA

    Sunnyvale, CA
    2 days ago
  •  ...Technical Lead Manager Applied Intuition builds the software infrastructure for autonomous vehicles across passenger cars, trucking, mining...  .... You will lead the team that trains, evaluates, and ships this...  ...someone who has: ~5+ years in ML/deep learning for perception... 
    Training
    Odd job
    For contractors
    For subcontractor

    Applied Compute

    Sunnyvale, CA
    1 day ago
  • $281k - $356k

     ...Software Technical Lead Manager, Test Automation Infrastructure Waymo is an autonomous driving technology company with the mission to be the world's most...  ...including exact work location, experience, relevant training and education, and skill level. Your recruiter can... 
    Training
    Full time
    Remote work

    Waymo

    Mountain View, CA
    3 days ago
  • $139.9k - $274.8k

     ...AI, we are on a mission to train the world's most capable AI...  ...and product deployment. Our Infrastructure team is responsible for...  ...motivated and detail-oriented Technical Program Managers to help bring our vision to...  ...scale infrastructure for AI/ML workloads. Have... 
    Training
    Ongoing contract
    Work at office
    Local area

    Microsoft Corporation

    Mountain View, CA
    1 day ago
  •  ...Cerebras to deliver industry-leading training and inference speeds...  ...run large-scale ML applications, without the hassle of managing hundreds of GPUs or TPUs...  .... This is a highly technical, execution-focused TPM...  ...AI Cloud Infrastructure & Operations Network... 
    Training

    CEREBRAS SYSTEMS INC.

    Sunnyvale, CA
    1 day ago
  • $281k - $356k

     ...environments for testing and training the Waymo Driver. Your...  ...of machine learning (ML) engineers, software...  ...scientists. We develop industry-leading simulation solutions...  ...Staff Engineering Manager. You will:...  ...Robotics, or a related technical field, or equivalent practical... 
    Training
    Full time
    Remote work

    Waymo

    Mountain View, CA
    3 days ago
  •  ...Cerebras to deliver industry-leading training and inference speeds and...  ...effortlessly run large-scale ML applications, without the hassle of managing hundreds of GPUs or TPUs....  ...agentic computation. As an Infrastructure Hardware Technical Program Manager (Server and... 
    Training

    CEREBRAS SYSTEMS INC.

    Sunnyvale, CA
    1 day ago
  • $238k - $302k

     ...Staff Technical Lead Manager, ML Sensor Validation Waymo is an autonomous driving technology company with the mission to be the world's most...  ...Build a data flywheel to automatically generate suitable training data for rare instance sensor degradation events using VLMs... 
    Training
    Full time
    Work at office
    Immediate start
    Remote work

    Waymo

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

     ...Staff Technical Lead Manager, Behaviors Waymo is an autonomous driving technology company with the...  ...eval workflows and building metrics for ML models The expected base salary...  ...exact work location, experience, relevant training and education, and skill level. Your... 
    Training
    Full time
    Remote work

    Waymo

    Mountain View, CA
    1 day ago
  •  ...Technical Lead, Evaluation Infrastructure Mountain View, California (HQ) Who We Are Nuro is a self-driving technology company on a mission to make...  ...insight. Our metrics stack spans both heuristic and ML-based approaches, covering everything from low-level component... 
    Temporary work
    Work experience placement

    Nuro

    Mountain View, CA
    3 days ago
  • $235.03k - $352.29k

     ...Technical Lead Manager, Autonomy Evaluation and Intelligence Mountain View, California (HQ)...  ...measure the cognitive performance of the ML models powering our self-driving car's...  ...cross-functionally with Autonomy and Infrastructure engineers to set a roadmap that unifies... 

    Nuro

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

     ...You will: Lead, mentor, and grow a team...  ...Define and execute the technical roadmap of the data pipeline...  ..., storage, and query infrastructure for Waymo driver...  ...experience in engineering management or tech leadership,...  ...experience, relevant training and education, and skill... 
    Training
    Full time
    Remote work

    Waymo

    Mountain View, CA
    4 days ago
  • $198k - $235.5k

     ...gives organizations the infrastructure to govern scale and customize...  ...a Senior Infrastructure Technical Program Manager (TPM) to lead large-scale cross-...  ...partner closely with Infra AI/ML and Platform Engineering...  ...evolve our ML pipelines model training infrastructure and LLM... 
    Training
    Work at office
    Home office
    Flexible hours

    Glean

    Mountain View, CA
    16 days ago
  • $281k - $356k

     ...Embedded Software Technical Lead & Manager, Simulation Waymo is an autonomous driving technology...  ...for expansion of test & sim infrastructure to support new products and Waymo's...  ...work location, experience, relevant training and education, and skill level. Your... 
    Training
    Full time
    Remote work

    Waymo

    Mountain View, CA
    2 days ago
  • $332k - $421k

     ...Principal Software Engineer, ML Flywheel Technical Lead Waymo is an autonomous driving technology...  ...to curation, (auto-)labeling, model training and evaluation, all the way to model...  ...in close collaboration with Waymo's infrastructure, modeling and evaluation teams. They... 
    Training
    Full time
    Remote work

    Waymo

    Mountain View, CA
    4 days ago
  • $235.03k - $352.29k

     ...Technical Lead, Behavior & Triage Labeling Mountain View...  ...technology. In an ML-first system, the overall...  ...and diversity of its training and evaluation data....  ..., supporting infrastructure, and a suite of data...  ...monitoring and incident management Experience applying... 
    Training

    Nuro

    Mountain View, CA
    2 days ago
  • $139.9k - $274.8k

     ...Microsoft AI, we are on a mission to train the world's most capable AI frontier...  ...highly motivated and detail-oriented Technical Program Managers to help bring our vision to life. We...  ...team. Work closely with teams on infrastructure, data engineering, pre-training, post... 
    Training
    Ongoing contract
    Work at office
    Local area

    Microsoft Corporation

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

    RadixArk

    Palo Alto, CA
    1 day ago
  • $132.1k - $165.1k

     ...Rivian Autonomy Technical Program Manager Rivian is on a mission to keep...  ...Pose (heavily reliant on ML), Planning & Controls, Infrastructure & Simulation (critical for ML model training/validation), Systems Definition...  .... Responsibilities Lead programs consisting of... 
    Training
    Full time
    Contract work
    Local area

    Rivian

    Palo Alto, CA
    3 days ago
  •  ...Engineer, AI Compute Infrastructure Los Angeles,...  ...models—from multimodal training data pipelines to...  ...for launching and managing massive,...  ...Experience with core ML frameworks such as...  ...Science or a related technical field. Demonstrated Tech Lead experience, driving... 
    Training
    Full time

    HeyGen

    Palo Alto, CA
    2 days ago
  • $118k - $390k

     ...Engineer within the Autopilot AI Infrastructure team, you will work on...  ...research team is designing to train on very large amounts of data...  ...with the supercomputing team managing the training cluster to maintain...  ...Experience working with ML training frameworks (ideally... 
    Training
    Hourly pay
    Full time
    Temporary work
    Flexible hours

    Tesla

    Palo Alto, CA
    4 days ago
  • $220k - $320k

     ...understanding, using, and risk-managing foundation models. Our...  ...edge foundation model training, alongside world-class...  ...The Role As the Technical Program Manager for...  ...Data Scientists, and Infrastructure Engineers. You must...  ...familiarity with the ML training lifecycle. You... 
    Training
    Visa sponsorship
    Flexible hours

    Institute of Foundation Models

    Sunnyvale, CA
    1 day ago
  • $109k - $160k

     ...GPU Infrastructure Software Engineer Sunnyvale, CA CoreWeave...  ...with confidence. Trusted by leading AI labs, startups, and...  ...infrastructure performance with deep technical expertise to accelerate...  .... Experience with AI/ML infrastructure and training / inference. The... 
    Training
    Permanent employment
    Temporary work
    Casual work
    Work at office
    Remote work
    Flexible hours

    CoreWeave

    Sunnyvale, CA
    2 days ago
  • $153k - $222k

     ...creating the digital infrastructure needed to bring intelligence...  ...our employees to manage their schedules responsibly...  ...pipelines and ML engineers that want to...  ...(dataset generation, training frameworks, compute, evaluation...  ...take ownership over technical and product decisions,... 
    Training
    Full time
    For contractors
    For subcontractor
    Casual work
    Work at office
    Remote work
    Day shift

    Applied Intuition

    Sunnyvale, CA
    16 hours ago
  • $157k - $235k

     ...Snap Engineering teams build fun and technically sophisticated products that reach...  ...a critical role in scaling our ML Infrastructure, optimizing training and inference systems, and driving...  ...serving Build comprehensive data management systems for scalable data... 
    Training
    Live in
    Work at office
    Local area

    Snapchat

    Palo Alto, CA
    16 hours ago
  • $159.96k - $191.46k

     ...Mobile Infrastructure Technical Manager - IT Business Affairs: University IT (UIT), Stanford, California...  .... This role will primarily focus on leading Mobile Infrastructure team,...  ...learning. Develop and implement training programs to enhance the skills and knowledge... 
    Training
    Full time
    Contract work
    Relocation
    Night shift
    2 days per week
    3 days per week

    Stanford University

    Stanford, CA
    5 days ago
  • $237k - $261k

     ...Staff Technical Program Manager - Cluster Orchestration & Applied Training CoreWeave is The Essential Cloud...  ...confidence. Trusted by leading AI labs, startups,...  ...CoreWeave combines superior infrastructure performance with deep...  ...within our AI/ML Platform Services organization... 
    Training
    Temporary work
    Casual work
    Work at office
    Remote work
    Flexible hours

    CoreWeave

    Sunnyvale, CA
    4 days ago
  • $159.4k - $245k

     ...Job Description Staff Technical Program Manager, Embodied AI At...  ...You translate complex, ML-heavy development...  ...initiatives you will own or lead within the Embodied...  ...execution of model training, evaluation, and performance...  ...with large-scale ML infrastructure, including... 
    Training
    Local area
    Work from home
    Relocation
    Relocation package
    Flexible hours

    General Motors

    Sunnyvale, CA
    16 hours ago
  •  ...Cloud Infrastructure Engineer At Rhoda AI, we're building the next generation...  ...infrastructure that collects training data, keeps our robots...  ...practices around security, cost management, and scalability...  ...building infrastructure for ML workloads — GPU cluster management... 
    Training

    Rhoda ai

    Palo Alto, CA
    2 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!