Job Description
Nuro is seeking a Staff Software Engineer to optimize machine learning infrastructure, focusing on improving training efficiency and input pipelines for self-driving model development, enabling faster and more reliable autonomous vehicle technology.
Key Responsibilities
- Collaborate with ML practitioners and infrastructure teams to integrate optimized input pipelines into workflows
- Detect, diagnose, and resolve performance bottlenecks in training, evaluation, and model distillation workflows
- Optimize training performance, resource utilization, and ensure reproducible model training outcomes
- Enhance input data pipelines to maximize runtime goodput and minimize idle cycles of accelerators
- Champion best practices for robust, reproducible, and debuggable ML experimentation
Requirements
- A B.S., M.S., or Ph.D. in Computer Science, Electrical Engineering, or related technical field or equivalent experience.
- At least 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.
- Highly proficient in C++, distributed systems, and ML framework internals such as NCCL, Horovod, DeepSpeed, or Ray.
Benefits & Perks
Compensation/salary range between $235,030 and $352,290 depending on experience and qualifications
Annual performance bonus
Equity
Competitive benefits package
Ready to Apply?
Join Nuro and make an impact in renewable energy
Stay Updated on Sustainability Jobs
Get the latest renewable energy jobs and career tips delivered to your inbox.
Job Alerts
Get notified about new sustainability jobs
More at Nuro
More jobs at Nuro
Embedded Software Engineer Intern
Nuro
NEW
Mountain View
Full Time
14h
Senior Project Manager, Software Operations
Nuro
NEW
Mountain View
Full Time
2d
$132k-198k
Safety Governance Standards Lead
Nuro
NEW
Mountain View
Full Time
2d
$194k-291k
More jobs in Mountain View, California
Staff Software Engineer - Data Curation
Aurora
Mountain View
Full Time
Nov 25
$189k-303k
Senior Software Engineer, Visualization
Aurora
Mountain View
Full Time
Dec 24
$139k-223k
Senior Software Engineer, Behavior Simulation
Aurora
Mountain View
Full Time
Oct 30
$139k-223k