GrepJob
Nuro

Software Engineer, ML Infrastructure, Optimization

Nuro
Apply
about 3 hours ago
Mountain View, CA, USAMid Level / Senior
H1B Sponsor

Base Salary

$160k - $241k/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

  • 2+ 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.
  • A passion for advancing robotics for everyday life.

Benefits

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

Tech Stack

C++KerasPythonPyTorchTensorFlow

Categories

AI & MLBackendData Engineering