Senior/Staff ML Performance Engineer, Low-Precision Training & Model Quantization jobs in United States
cer-icon
Apply on Employer Site
company-logo

Nuro · 2 months ago

Senior/Staff ML Performance Engineer, Low-Precision Training & Model Quantization

Nuro is a self-driving technology company on a mission to make autonomy accessible to all. They are seeking an experienced ML Performance Engineer to drive the adoption of quantization techniques and improve model training efficiency for their ML infrastructure.

Artificial Intelligence (AI)Autonomous VehiclesRoboticsTransportation
check
H1B Sponsor Likelynote

Responsibilities

Staying ahead of emerging research and evaluating new methods
Implementing quantized training methods (e.g., AWQ, AQT, GPTQ) for new and existing self-driving models
Leading the design and implementation of efficiency initiatives for model training, including low-bit quantization, and pruning for both research and production workloads
Championing and implementing tools and approaches to pinpoint root-causes for possible model quality and accuracy regressions when training at lower precisions
Collaborating cross-functionally with research, infrastructure, and product teams balancing accuracy, latency, and resource constraints

Qualification

Quantization methodsML infrastructureDistributed trainingModel compression techniquesCalibration strategiesResearch experienceCollaboration skills

Required

3+ 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
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

Preferred

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)

Benefits

Annual performance bonus
Equity
Competitive benefits package

Company

Nuro is a robotics company specializing in the development of autonomous driving technologies.

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
Company data provided by crunchbase