Senior/Staff Software Engineer, ML Infrastructure, Optimization
Nuro
about 1 year ago
Mountain View, CA, USA
Senior / Staff+
H1B Sponsor
Base Salary
$167k - $303k/yr
Responsibilities
- Optimize Nuro’s autonomy stack using advanced optimization techniques.
- Collaborate with autonomy engineers to validate and deploy large language models.
- Develop and maintain the FTL model compiler framework.
- Write high-quality software to enhance vehicle navigation safety.
- Work with machine learning experts to design end-to-end ML solutions.
Requirements
- 3+ years of experience in ML optimization infrastructure.
- Familiarity with ML optimization techniques like quantization and pruning.
- Experience with GPU ML compilers and runtimes.
- Proficient in Python, with working knowledge of C++ and CUDA.
- Experience with deep learning frameworks such as PyTorch, Jax, TensorFlow, or Keras.
- Passion for advancing robotics technology for everyday life.
Benefits
- Eligible for an annual performance bonus.
- Equity options available.
- Competitive benefits package.
Tech Stack
C++KerasPythonPyTorchTclTensorFlow
Categories
AI & MLData Engineering