Job Description

Rivian is seeking a Staff Software Engineer in ML Acceleration and Performance to join the Perception team, focusing on optimizing and deploying machine learning algorithms for safety-critical self-driving features in electric vehicles.

Requirements

MS or Ph.D. in Computer Science, Electrical, Mechanical, Aerospace Engineering or a related field.
Good understanding of deep learning, heterogeneous computing, ML accelerators and compilers with 7+ years of industrial experience.
Research and development experience in model compression and neural architecture search techniques.
Experience defining compute architecture for efficient Deep learning inferencing.
Strong Python programming background and knowledge of at least one framework like PyTorch, TensorFlow or MXNet.
Experience implementing inference logic using low-level subroutines like BLAS, CUDA Kernels or C++.
Familiarity with ONNX runtimes and DL compilers like XLA or Glow.

Responsibilities

Develop, optimize and deploy ultra-low latency Deep learning/Machine Learning algorithms for Rivian ADAS and Autonomy use cases.
Research state of the art model compression and efficient model design techniques.
Collaborate with low-level software and hardware architecture teams to optimize ML models on embedded platforms.