Senior/Staff Machine Learning Engineer, Training Runtime Performance jobs in United States
cer-icon
Apply on Employer Site
company-logo

Nuro · 4 days ago

Senior/Staff Machine Learning Engineer, Training Runtime Performance

positionMountain View, California (HQ)
timeFull-time
remoteOnsite
senioritySenior Level, Lead/Staff
money$235K/yr - $352K/yr
date4+ years exp

Nuro is a self-driving technology company on a mission to make autonomy accessible to all. They are seeking a highly experienced Staff Software Engineer to join their ML Infrastructure team, focusing on optimizing training runtime efficiency and input pipelines for model training, evaluation, and distillation workloads.

Artificial Intelligence (AI)Autonomous VehiclesRoboticsTransportation
check
H1B Sponsor Likelynote

Responsibilities

Collaborate with ML practitioners and other infrastructure teams to understand their needs and integrate optimized input pipelines seamlessly into their workflows
Detect, diagnose, and resolve performance bottlenecks across training, eval, and model distillation workflows
Optimize training performance, resource utilization, and ensure consistent, reproducible model training outcomes
Optimize input data pipelines to increase runtime goodput, ensuring accelerators maximize their "time on task" and minimize idle cycles
Champion best practices for robust, reproducible, and debuggable ML experimentation

Qualification

ML infrastructureDistributed trainingPythonC++Model distillationPerformance optimizationDebuggingProfilingOpen source contributionsCollaboration

Required

B.S./M.S./Ph.D. in Computer Science, Electrical Engineering, or related technical field (or equivalent experience)
4+ years of professional experience in ML infrastructure, distributed training, or ML systems engineering, scaling models on multi-node, multi-accelerator clusters
Understanding of training, evaluation, and distillation workflows for billion-parameter models
Expert-level knowledge in distributed systems and (remote) Python
Strong skills in profiling, debugging, and optimizing quantized workloads
Experience with ML compilers and strategies to reduce startup overhead
Familiarity with model distillation and efficient inference workflows

Preferred

Previous contributions to open source ML infra projects or research publications in ML systems
Hands-on experience with Foundation Model infrastructure
Highly proficient in C++, distributed systems, ML framework internals (e.g., NCCL, Horovod, DeepSpeed, Ray)

Benefits

Annual performance bonus
Equity
Competitive benefits package

Company

Nuro

twittertwittertwitter
Glassdoor3.8
company-logo
Nuro is a robotics company specializing in the development of autonomous driving technologies.
dateFounded in 2016
location
Mountain View, California, USA
people-size501-1000 employees
web-link
https://nuro.ai

H1B Sponsorship

Nuro has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (50)
2024 (67)
2023 (91)
2022 (91)
2021 (68)
2020 (55)

Funding

Current Stage
Late Stage
Total Funding
$2.34B
Key Investors
Disney AcceleratorTiger Global ManagementT. Rowe Price
2025-08-21Series E· $97M
2025-04-09Series E· $106M
2024-04-06Secondary Market

Leadership Team

leader-logo
Jiajun Zhu
Co-Founder and Co-CEO
linkedin
leader-logo
Dave Ferguson
Co-Founder
linkedin

Recent News

Business Insider
I'm at CES in Las Vegas to check out the latest in autonomous driving. Here's what I've learned so far.
2026-01-07
IndiaTimes
CES 2026: The technologies defining the year ahead
2026-01-07
Business – Baltimore Sun
The coolest technology from Day 1 of CES 2026
2026-01-07
Company data provided by crunchbase