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

Nuro · 3 hours ago

Senior/Staff Software Engineer, Machine Learning Infrastructure

Nuro is a self-driving technology company focused on making autonomy accessible to all. The role involves designing and maintaining tools for the model development lifecycle, ensuring efficient deployment and evaluation of machine learning services.

Artificial Intelligence (AI)Autonomous VehiclesRoboticsTransportation
check
H1B Sponsor Likelynote

Responsibilities

Implement tools to track the model development lifecycle for an efficient deployment and evaluation process
Develop issue detection and alerting mechanisms for critical ML services (training jobs, data pipelines, model deployments) to quickly detect issues and ensure high uptime
Maintain observability dashboards to track model performance, data quality, and system metrics
Champion best practices for robust, reproducible, and debuggable ML experimentation
Collaborate with cross-functional teams (perception, behavior, mapping, simulation teams, etc.) to identify infrastructure needs and integrate solutions across various domains for end-to-end ML solutions
Thoughtfully balance innovative exploration with practical considerations

Qualification

PythonC++ML development lifecycleDistributed systemsPytorchTensorFlowCollaborationProblem-solving

Required

2+ years of relevant work experience or an equivalent experience in Masters/PhD with 2+ years of relevant experience
Strong coding skills in Python or C++
Exposure to the ML development life cycle and ML models
Experience profiling and optimizing performance bottlenecks in distributed systems
Eager to overcome obstacles and accelerate development make systems work better for the people who use them

Preferred

Work experience designing, implementing, operating, or maintaining feature and model stores
Previous contributions to open source ML Infrastructure projects or research publications in ML systems
Experience with Pytorch, JAX, or TensorFlow

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