As a Staff Software Engineer on the Perception team at Rivian, you will focus on developing advanced machine learning algorithms that enhance safety-critical self-driving features in electric vehicles. The role requires a deep understanding of machine learning infrastructure for training and inference of large models. You will work on optimizing performance and latency for deep learning workloads on NVIDIA GPU systems. The team values diverse backgrounds and a shared passion for the outdoors and sustainability.

• PhD in Computer Science, Computer Engineering, Electrical Engineering, or equivalent industry experience.
• Deep knowledge of PyTorch.
• Knowledge of model training frameworks such as PyTorch Lightning or Ray.
• In-depth knowledge of transformer architecture and methods to accelerate training and inference.
• Experience with large scale distributed training of models.
• A track record of profiling models and improving training and inference speed.

• Optimize the performance of deep learning training workloads on NVIDIA GPU systems at scale.
• Optimize the latency of model inference and pre- and post-processing on onboard systems.
• Design, train, and deploy large deep learning models leveraging vast amounts of labeled and unlabeled data.

• Employees at Rivian are often offered comprehensive health, dental, and vision insurance, a 401(k) with company match, ESPP and potential RSU grants, performance bonuses, and generous time off including paid parental and sick leave. Perks can include 24/7 mental-health coaching and therapy, Hinge Health for Anthem enrollees, tuition assistance and professional development, plus on-site food and a wide range of employee discounts on items like computers, mobile phones, home loans, and pet care.