Job Description
The role involves building and optimizing core machine learning infrastructure for autonomous vehicles at Nuro, including model deployment, optimization techniques, and collaboration with engineering teams to advance self-driving technology.
Key Responsibilities
- Design, build, and deploy core infrastructure components for machine learning model lifecycle
- Optimize autonomy stack using techniques like quantization, distillation, and model compression
- Develop and maintain a model compiler framework (FTL)
- Write robust software to ensure vehicle navigation safety
- Collaborate with ML domain experts to design and implement end-to-end learned ML solutions
Requirements
- At least 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 programming language.
- Working experience with C programming language.
- Working experience with CUDA.
- Experience with deep learning frameworks such as PyTorch, Jax, TensorFlow, or Keras.
Benefits & Perks
Compensation/salary range between $193,930 and $352,290
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
2d
Senior Project Manager, Software Operations
Nuro
Mountain View
Full Time
3d
$132k-198k
Safety Governance Standards Lead
Nuro
Mountain View
Full Time
3d
$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