Staff Software Engineer, ML Acceleration and Performance

Rivian • Palo Alto, CA

69d • $179,000 - $285,000

About The Position

As a Staff Software Engineer, ML Acceleration and Performance, you will be a member of the Perception team at Rivian, which develops advanced machine learning algorithms that directly impact safety critical self-driving features of our category defining vehicles.

Requirements

• MS. or Ph.D. in Computer Science, Electrical, Mechanical, Aerospace Engineering or a related field.
• Good understanding of the fundamentals of deep learning, heterogenous computing, ML accelerators and compilers with 7+ years of industrial experience.
• Good team player with great communication skills to drive cross functional efforts in a fast-paced development environment.
• Research and development experience in one or more of the following areas: model compression and neural architecture search techniques knowledge distillation, pruning, quantization and quantization aware training.
• Experience defining compute architecture for efficient Deep learning inferencing.
• Capability to understand hardware spec documents and performance profiling tools.
• Strong Python programming background and in-depth knowledge of at least one framework amongst PyTorch, TensorFlow or MXNet.
• Experience implementing inference logic from first principles using low level subroutines like BLAS, CUDA Kernels or C++ natively.
• 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 and enable the team to leverage these across a wide range of customer facing features.
• Collaborate with the low-level software and hardware architecture teams to characterize the in-house ML models on our embedded platforms and optimize the models subject to the on device compute and memory constraints.

Benefits

• Robust medical/Rx, dental and vision insurance packages for full-time employees, their spouse or domestic partner, and children up to age 26.
• Coverage is effective on the first day of employment.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Industry: Transportation Equipment Manufacturing
• Education Level: Master's degree
• Number of Employees: 5,001-10,000 employees