Software Engineer, ML Infrastructure, Optimization

Nuro Nuro · Robotics · CA · Software

Software Engineer focused on ML infrastructure and optimization for Nuro's self-driving robots. Responsibilities include optimizing ML models using techniques like quantization and pruning, developing a model compiler framework, and collaborating with autonomy engineers to deploy optimized models.

What you'd actually do

  1. Optimize Nuro’s autonomy stack with cutting-edge optimization techniques like quantization, low precision inference, and model pruning.
  2. Work with autonomy engineers to optimize, validate, and deploy large language models.
  3. Develop and maintain a world-class model compiler framework, [FTL](https://medium.com/nuro/ftl-model-compiler-framework-d6b85c670f67).
  4. Write robust, high-quality software to increase our confidence in our vehicle’s ability to navigate safely on-road.
  5. Collaborate closely with machine learning domain experts and engineers across behavior, perception and mapping to design and implement end-to-end learned ML solutions.

Skills

Required

  • ML optimization infrastructure
  • quantization
  • pruning
  • ML compilers
  • GPU ML compilers & runtimes
  • Python
  • C++
  • CUDA
  • deep learning frameworks (PyTorch, Jax, Tensorflow, Keras)

Nice to have

  • LLMs

What the JD emphasized

  • ML optimization infrastructure
  • ML optimization techniques such as quantization and pruning
  • ML compilers
  • GPU ML compilers & runtimes
  • Python
  • C++
  • CUDA
  • deep learning frameworks (like PyTorch, Jax, Tensorflow, Keras)

Other signals

  • optimize ML models
  • deploy models on fleet
  • ML infrastructure