Staff Machine Learning Engineer, Autonomous Driving Systems

Rivian Automotive; City Of London; On-site; GBP 120,000 - 180,000; Full time

Job summary

Rivian is seeking a Staff Machine Learning Engineer to advance end-to-end autonomous driving models that unify perception, prediction, and planning in a scalable learning system. The role focuses on scaling models from vast sensory data to enable autonomous driving and requires leadership across ML projects.

You will work with the Autonomy team to architect and deploy the Large Driving Model, collaborating with perception, planning, simulation, and ML infrastructure to meet deployment

Qualifications

• BS, MS, or PhD in Computer Science, Robotics, or related field.
• 5+ years 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 with 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

• Developing technical strategy and architecture for end-to-end autonomous driving model.
• Developing multi-modal, multi-task transformer-based systems that support closed-loop autonomy.
• Building training and evaluation pipelines at scale across petabytes of real-world and simulated driving data.
• Collaborating with cross-functional teams across perception, planning, simulation, and ML infrastructure.
• Driving alignment between model capabilities and real-world deployment constraints (latency, robustness, validation).
• Publishing internal technical guidance and mentoring engineers across autonomy ML.

Skills

Foundation models

Self-supervised learning

World models

Python

C++

Transformer models

Autonomy ML leadership

Education

B.S., M.S., or Ph.D. in Computer Science, Robotics, or a related field

Job description

Rivian is seeking a Staff Machine Learning Engineer to advance end-to-end autonomous driving models that unify perception, prediction, and planning in a scalable learning system. The role focuses on scaling models from vast sensory data to enable autonomous driving and requires leadership across ML projects.

You will work with the Autonomy team to architect and deploy the Large Driving Model, collaborating with perception, planning, simulation, and ML infrastructure to meet deployment