Software Engineer, ML Infrastructure, Optimization jobs in United States
cer-icon
Apply on Employer Site
company-logo

Nuro · 7 hours ago

Software Engineer, ML Infrastructure, Optimization

Nuro is a self-driving technology company on a mission to make autonomy accessible to all. The role involves optimizing the autonomy stack and developing infrastructure for machine learning solutions, working closely with various teams to enhance the capabilities of self-driving robots.

Artificial Intelligence (AI)Autonomous VehiclesRoboticsTransportation
check
H1B Sponsor Likelynote

Responsibilities

Optimize Nuro’s autonomy stack with cutting-edge optimization techniques like quantization, low precision inference, and model pruning
Work with autonomy engineers to optimize, validate, and deploy large language models
Develop and maintain a world-class model compiler framework, FTL
Write robust, high quality software to increase our confidence in our vehicle’s ability to navigate safely on-road
Collaborate closely with machine learning domain experts and engineers across behavior, perception and mapping to design and implement end-to-end learned ML solutions

Qualification

PythonC++Deep learning frameworksML optimization techniquesGPU ML compilersCollaborationMotivationAdaptability

Required

Highly productive, motivated, and comfortable with new, unfamiliar environments
Proficient in Python and C++
Working experience deep learning frameworks (like PyTorch, Jax, Tensorflow, Keras)
You are passionate about accelerating the benefits of robotics for everyday life

Preferred

Experience with ML optimization techniques such as quantization and pruning, and ML compilers
Experience maintaining, profiling, and optimizing GPU ML compilers & runtimes
Experience writing custom ML operators in CUDA and Triton Language

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