Job Description
A role focused on building and optimizing the 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 autonomous systems using techniques like quantization, distillation, and model compression
- Collaborate with engineers to validate, optimize, and deploy large language models
- Develop and maintain a model compiler framework (FTL)
- Write robust software to ensure vehicle navigation safety
- Work with 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 and working experience with C and CUDA.
- Working experience with deep learning frameworks like PyTorch, Jax, Tensorflow, or Keras.
Benefits & Perks
Compensation/salary range: $193,930 to $352,290 (base pay)
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
Jobs in Mountain View, California
More jobs at Nuro
Senior Staff Software Engineer, ML Data
Nuro
NEW
Mountain View
Full Time
14h
$194k-352k
Mission Operations Coordinator, Houston Overnight Shift
Nuro
Depot 2
Full Time
3d
$56k-75k
Mechanical Engineer, Autonomous Vehicle
Nuro
Mountain View
Full Time
3d
$120k-180k
More jobs in Mountain View, California
Test Triage Analyst I
Aurora
NEW
Mountain View
Full Time
2d
$76k-114k
Senior Photonics Engineer
Aurora
NEW
Mountain View
Full Time
2d
$150k-240k
Senior Product Quality Engineer
Aeva
NEW
Mountain View
Full Time
14h