Job Description
The role involves developing and optimizing scalable machine learning infrastructure and distributed training frameworks for deep learning models, supporting Nuro's autonomous driving technology.
Key Responsibilities
- Research and develop distributed training frameworks and strategies for deep learning models
- Optimize model training speed by improving TensorFlow, Keras, PyTorch, and CUDA kernel implementations
- Engineer tools to profile and monitor model training performance and detect training issues
Requirements
- 5 years of relevant work experience, or an equivalent experience in PhD.
- Understanding of machine learning models and ML development life cycle.
- Experience in building a cloud-based distributed training platform that fully supports data and model parallelism.
- Experience investigating and optimizing training performance bottlenecks for deep learning models.
- Research and develop new distributed training frameworks and strategies to support training deep learning models with growing sizes.
- Improve model training speed by optimizing TensorFlow, Keras, PyTorch, and Cuda kernel implementation.
- Engineer advanced tools to profile and monitor model training performance across all teams, to detect and triage training problems.
Benefits & Perks
Base salary range between $193,930 and $352,290 depending on experience and qualifications
Annual performance bonus
Equity options
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
13h
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