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
  • $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
    2 days 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
    4 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
    4 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... 
    Suggested
    Temporary work
    Work experience placement
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    23 days 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... 
    Suggested
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    9 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
    16 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
    3 days 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
    23 hours 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
    11 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
  • $165k - $242k

     ...AI with confidence. Trusted by leading AI labs, startups, and global...  ..., CoreWeave combines superior infrastructure performance with deep technical expertise to accelerate breakthroughs...  ...Experience with AI/ML infrastructure and training / inference The base salary... 
    Training
    Permanent employment
    Temporary work
    Casual work
    Work at office
    Flexible hours

    CoreWeave

    Sunnyvale, CA
    29 days ago
  •  ...lakehouses, the traditional ML approach has failed them with...  ...methods. Your Mission As an Infrastructure Engineer at Kumo, you'll...  ...Stanford, you'll solve complex technical challenges that few engineers...  ...-scale data, complex ML training, and reliable inference—all while... 
    Training

    Kumo.AI

    Mountain View, CA
    more than 2 months ago
  • $160k - $225k

     ...best data and AI infrastructure platform so our customers...  ...their business. Training and customizing...  ...(AIR) is our managed platform for large...  ...will help shape the technical direction for AIR,...  ...production training jobs. Lead end‑to‑end...  ...performance computing, or ML systems.... 
    Training
    Local area

    Databricks

    Mountain View, CA
    4 days ago
  • $193.93k - $291.15k

     ...Price, and other leading investors About...  ...expertise in large-scale infrastructure, workload...  ...processing to join our ML Infrastructure...  ...executing the technical strategy for automated...  ...feature management to accelerate the...  ...scale distributed training. Data & ETL:... 
    Training
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    4 days ago
  •  ...JOB DESCRIPTION As a Lead Software Engineer of Software...  ...Team, you lead multiple technical areas, manage the activities of multiple...  ...capabilities, and skills Formal training or certification on...  ...metrics, log aggregation, and infrastructure monitoring.  ~ Partner with... 
    Training
    Bank staff

    J.P. Morgan

    Palo Alto, CA
    1 day ago
  • $285k - $352k

     ...Head Of Technical Talent Acquisition Waymo is...  ...Engineering Recruiting to lead our technical talent...  ...team of recruiting managers and ICs, and...  ...organization (Software, Infrastructure, Security, AI/ML) in line with company...  ...experience, relevant training and education, and... 
    Training
    Full time
    Contract work
    For contractors
    Remote work

    Waymo

    Mountain View, CA
    23 hours ago
  • $177k - $237k

     ...confidence. Trusted by leading AI labs, startups,...  ...combines superior infrastructure performance with deep technical expertise to...  ...You'll Do: The AI/ML TPM team owns delivery...  ...Technical Program Manager, you will lead complex...  ...validation for training and inference workloads... 
    Training
    Permanent employment
    Temporary work
    Casual work
    Work at office
    Flexible hours

    CoreWeave

    Sunnyvale, CA
    20 days ago
  •  ...and experienced Principal AI/ML and Gen AI Engineer to join...  ...focused on scaling infrastructure and platforms from a serving...  ...for data processing, model training, and deployment. Utilize...  ...efficient model lifecycle management. Lead the productionalization of... 
    Training

    J.P. Morgan

    Palo Alto, CA
    20 days ago
  •  ...Synthetic Data Group (SDG) Cloud Infrastructure team at Apple is responsible...  ...services and pipelines to process and manage large volumes of data, enabling cutting-edge ML applications. This role...  ...and play a key role in setting technical direction. You’ll have the opportunity... 

    Apple

    Sunnyvale, CA
    4 days ago
  •  ...performing team that delivers infrastructure and performance...  ...companies. As a Lead Infrastructure Engineer...  ...software, applications, and technical processes within the...  ...expectations. Manages stakeholder relationships...  ...and skills Formal training or certification on Infrastructure... 
    Training
    Permanent employment

    J.P. Morgan

    Palo Alto, CA
    15 days ago
  •  ...reliability.  As a Senior Lead Site Reliability...  ...Chase within the Infrastructure Platforms and...  ...seeking advice on technical and business...  ...skills Formal training or certification on...  ...Familiarity with AI/ML model building, deployment...  ..., and lifecycle management using frameworks... 
    Training

    J.P. Morgan

    Palo Alto, CA
    4 days ago
  • $160.36k - $240.54k

     ...Rowe Price, and other leading investors About the Role...  ...technology. In an ML-first system, the overall...  ...and diversity of its training and evaluation data....  ...scalable and reliable data infrastructure. This infrastructure...  ...complex systems Technical excellence: Ability and... 
    Training
    Work experience placement
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    16 days ago
  • $193.93k - $291.15k

     ...Price, and other leading investors. About...  ...generalists where ML and systems engineering...  ...ML Data Infrastructure Engineer, you will...  ...raw sensor data and manage complex data pipeline...  ...Platforms : Take formal technical ownership of our...  ...they enter the ML training loop. Cross-... 
    Training
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    29 days ago
  • $237k - $261k

     ...confidence. Trusted by leading AI labs, startups, and...  ...combines superior infrastructure performance with deep technical expertise to accelerate...  ...Staff Technical Program Manager to lead complex, cross...  ...Orchestration and Applied Training within our AI/ML Platform Services... 
    Training
    Permanent employment
    Temporary work
    Casual work
    Work at office
    Flexible hours

    CoreWeave

    Sunnyvale, CA
    24 days ago
  • $220k - $320k

     ...Job Description About the role Own the infrastructure that engineering depends on — Kubernetes...  ...platforms for chip-design and ML workloads. Work with chip-design and software...  ...build systems, ML-platform infrastructure (training / inference), or RAG / knowledge-platform... 
    Training
    H1b
    Visa sponsorship
    Work visa

    DensityAI

    Mountain View, CA
    18 days ago
  •  ...017, Wayve is the leading developer of Embodied...  ...to build the infrastructure to scale it. Your...  ...models Build and manage distributed data processing...  ...influence model training strategy...  ...deliver scientific and technical impact What you...  ...closely with ML researchers and contributing... 
    Training
    Full time
    Work at office
    Remote work
    Work from home
    Visa sponsorship
    Relocation package
    Flexible hours

    GrabJobs

    San Jose, CA
    23 hours ago
  • $99k - $232k

     ...to 60% The Opportunity As a Manager, Oracle Cloud Technical Lead, you will play a pivotal role in guiding...  ...software solutions, and provide training and support for seamless...  ...Publisher Utilizing Oracle Cloud Infrastructure to enhance system performance and... 
    Training
    Full time
    H1b

    PwC

    San Jose, CA
    1 day ago
  •  ...products. As a Senior Lead Software Engineer...  ...Corporate Sector, Infrastructure Platforms team,...  ..., and apply deep technical expertise and...  ...requirements. Monitor, manage, and optimize...  ...skills Formal training or certification on...  ...transformer architecture, ML training, and... 
    Training
    For contractors

    J.P. Morgan

    Palo Alto, CA
    1 day ago
  • $204k - $259k

     ...schedule and reports to an Engineering Manager. You will: Develop signals to...  ...coding in C++. Experience with ML. Experience with A/B experiment infrastructure. Experience building and validating...  ...location, experience, relevant training and education, and skill level. Your... 
    Training
    Full time
    Remote work

    Waymo

    Mountain View, CA
    31 minutes ago
  •  ...We’re hiring a Machine Learning Infrastructure Engineer to help build the backbone that trains, serves, and monitors the models...  ...launched model. You’ll work across the ML lifecycle: making training faster...  ...the stack and a clear path to lead projects and mentor others.... 
    Training
    Full time

    NewsBreak

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