Senior ML Infra Engineer for Autonomous Driving

Rivian is seeking a full-time Senior ML, AI Infra Engineer to architect and optimize the AI infrastructure for training very large foundation models and accelerating inference.

You will collaborate with software engineers, ML engineers and researchers to push state-of-art ML for autonomous driving, designing scalable training pipelines and robust deployment.

Applicants should have PhD, PyTorch expertise, CUDA/Triton experience, and proven distributed training capabilities.