Senior Staff Software Engineer, Distributed Training, Machine Learning Infrastructure
NuroPosted February 21, 2025
Mountain View, California
Electric Vehicles$167k - $303k
Full Time
Job Description
The role involves developing scalable machine learning infrastructure and distributed training solutions to enhance ML development at Nuro, focusing on optimizing training frameworks and improving model performance.
Responsibilities
- •Research and develop new distributed training frameworks and strategies for deep learning models.
- •Optimize Tensorflow, Keras, Pytorch, and Cuda kernel implementations to improve model training speed.
- •Engineer tools to profile and monitor model training performance and detect training problems.
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, which fully supports data and model parallelism.
- •Experience investigating and optimizing training performance bottlenecks for deep learning models.
Benefits
- •Compensation/salary range: between 167,200 and 303,050
- •Annual performance bonus
- •Equity
- •Competitive benefits package
Stay Updated on Sustainability Jobs
Get the latest renewable energy jobs and career tips delivered to your inbox.
More jobs at Nuro
Machine Learning Research Scientist, Behavior Planning and Prediction
Nuro
Jan 27
Machine Learning Research Scientist Generative Models for Behavior Modeling
Nuro
Jan 27
More jobs in Mountain View, California
Principal Hardware Test Automation Engineer
Aeva
Jan 27