Staff ML Infra Architect - Autonomous Training

Rivian; Palo Alto (CA); On-site; USD 207,000 - 258,000; Full time

Job summary

Rivian seeks a Staff Software Engineer to design and build the ML training platform powering autonomous driving models. You will shape the control plane that schedules jobs across clusters, the storage/I O layer, observability, and the training framework.

The role requires leading end-to-end in one or two areas while collaborating with model teams, Cloud Infrastructure, and Data Infrastructure. As an early member, you will help define the team's technical direction, operating model, and future

Qualifications

• 5+ years of software engineering experience, with substantial distributed-systems work in production.
• Strong programming skills in Python and experience with at least one additional language such as Go, Rust or C++.

Responsibilities

• Lead one or two platform areas end-to-end (job scheduling, storage I/O, observability, training framework) and contribute across others.
• Partner with ML model teams, Cloud Infrastructure, and Data Infrastructure teams to ship platforms engineers adopt.

Skills

Distributed systems

Python

Go

Rust

C++

Job description

Rivian seeks a Staff Software Engineer to design and build the ML training platform powering autonomous driving models. You will shape the control plane that schedules jobs across clusters, the storage/I O layer, observability, and the training framework.

The role requires leading end-to-end in one or two areas while collaborating with model teams, Cloud Infrastructure, and Data Infrastructure. As an early member, you will help define the team's technical direction, operating model, and future