Staff ML Engineer - Autonomous Driving AI Training & Inference

Rivian is seeking a Staff Engineer to lead ML training and infrastructure within the Perception team. You will help build state-of-the-art ML infrastructure for training and inference in large autonomous driving models, and optimize performance across training and inference pipelines.

You will define cross-team technical strategies, elevate engineering practices, and benchmark modern tools and frameworks to push the capabilities of our self-driving technologies.