Senior Staff Software Engineer, ML Infrastructure, Optimization
NuroPosted February 25, 2025
Mountain View, California
Electric Vehicles$167k - $303k
Full Time
Job Description
The role involves optimizing machine learning infrastructure for autonomous driving at Nuro, focusing on model training pipelines, inference optimizations, and developer productivity. The position requires collaboration with various teams to design, build, and deploy machine learning solutions for self-driving robots.
Responsibilities
- •Optimize Nuro's autonomy stack using optimization techniques like quantization, distillation, and model compression.
- •Collaborate with autonomy engineers to optimize, validate, and deploy large language models.
- •Develop and maintain a model compiler framework.
- •Write high-quality software to enhance vehicle navigation safety.
- •Design and implement end-to-end learned ML solutions in collaboration with machine learning experts.
Requirements
- •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, Keras.
Benefits
- •Compensation/salary range: between 167,200 and 303,050
- •Annual performance bonus
- •Equity
- •Competitive benefits package
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