Rivian is seeking a Staff Machine Learning Engineer to join their Autonomy team, focusing on the development of end-to-end models that integrate perception, prediction, and planning for autonomous driving. The ideal candidate will have a strong background in machine learning and software engineering, with a passion for solving complex problems in the automotive space. This role involves working with large-scale sensory data and requires collaboration across various teams to ensure effective deployment of machine learning models. Candidates should be excited about the challenges of scaling models for real-world applications.

• B.S., M.S., or Ph.D. in Computer Science, Robotics, or a related field
• 5+ 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

• Develop the technical strategy and architecture for an end-to-end autonomous driving model
• Create multi-modal, multi-task transformer-based systems that support closed-loop autonomy
• Build training and evaluation pipelines at scale using 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 such as latency and robustness
• Publish internal technical guidance and mentor engineers across autonomy ML

• 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.