Sr. Machine Learning Engineer, Perception

Job Description

Rivian is seeking a key member for their Perception team to develop advanced machine learning algorithms for self-driving features in Electric Adventure Vehicles.

Requirements

• B.S., M.S., or Ph.D. in Computer Science, Robotics, or a related field
• 7+ years of experience building and deploying large-scale ML systems
• Deep understanding of foundation models, self-supervised learning, and world models in robotics or simulation
• Strong software engineering background, with fluency in Python and C++
• Experience training and evaluating transformer models or end-to-end autonomous agents
• Familiarity with real-time inference systems and autonomous vehicle constraints
• Proven leadership in driving ML projects from research to production

Responsibilities

• Guide the architecture, implementation, and deployment of foundation models as learned world models
• Develop multi-modal, multi-task transformer-based systems for closed-loop autonomy
• Build training and evaluation pipelines at scale across petabytes of real-world and simulated driving data
• Collaborate with cross-functional teams across perception, planning, simulation, and ML infrastructure
• Drive alignment between model capabilities and real-world deployment constraints
• Publish internal technical guidance and mentor engineers across autonomy ML

Benefits

• Robust medical/Rx, dental and vision insurance packages for full-time and part-time employees, their spouse or domestic partner, and children up to age 26.