Job Description

Rivian is seeking a key member for the ML Compiler team to develop software tools for deep learning network inference on Rivian hardware platforms, collaborating closely with ADAS and hardware teams to optimize performance and support new hardware features.

Requirements

Degree in Computer Engineering or a related field
Excellent C/C++ and Python programming skills
Experience with various SOC platforms used for machine learning and ADAS
Strong understanding of deep learning software models
Experience in compiler pipeline development preferred
Experience working in aggressive design environments preferred

Responsibilities

Fast and accurate simulation models for Rivian Hardware Platform
Performance analysis and optimization of ML workloads
Work closely with ADAS and software teams to facilitate migration of software models in hardware