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

Nuro · 5 months ago

Senior/Staff Software Engineer, ML Infrastructure, Optimization

Nuro is a self-driving technology company focused on making autonomy accessible to all. The Senior/Staff Software Engineer in ML Infrastructure, Optimization will design, build, and deploy core infrastructure components in the machine learning model life cycle, optimizing Nuro's autonomy stack and collaborating with various teams to push the autonomous future forward.

Artificial Intelligence (AI)Autonomous VehiclesRoboticsTransportation
check
H1B Sponsor Likelynote

Responsibilities

Optimize Nuro’s autonomy stack with cutting-edge optimization techniques like quantization, distillation, and model compression
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

ML optimization infrastructureML optimization techniquesGPU ML compilersPythonC++CUDADeep learning frameworksPassion for roboticsCollaboration

Required

3+ years of relevant experience in ML optimization infrastructure
Experience with ML optimization techniques such as quantization and pruning, and ML compilers
Experience maintaining, profiling, and optimizing GPU ML compilers & runtimes
Proficient in Python and working experience with C++ and CUDA
Working experience deep learning frameworks (like PyTorch, Jax, Tensorflow, Keras)
Proficient in Python and working experience with C++

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