GrepJob
Nuro

Senior/Staff Software Engineer, ML Infrastructure, Optimization

Nuro
Apply
about 3 hours ago
Mountain View, CA, USASenior / Staff+
H1B Sponsor

Base Salary

$194k - $352k/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 closely with machine learning experts to implement 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.
  • Passionate about leveraging robotics for everyday life.

Benefits

  • Eligible for an annual performance bonus.
  • Equity options available.
  • Competitive benefits package.

Tech Stack

C++KerasPythonPyTorchTensorFlow

Categories