Senior AI Infra Engineer for Large-Scale ML & Autonomy

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

Job summary

Rivian is seeking a full-time Senior ML, AI Infra Engineer to design and operate state-of-the-art infrastructure for training large foundation models and accelerating inference for autonomous driving applications. You will collaborate with software engineers, ML engineers and researchers to push the boundaries of ML systems at scale.

The role emphasizes building robust training pipelines, optimizing performance, and deploying models across a growing fleet of vehicles.

Qualifications

• PhD in CS/CE/EE, or equivalent industry experience.
• Deep knowledge of PyTorch and large-scale ML systems.
• Experience with CUDA or Triton for custom ops.
• Knowledge of model training frameworks and distributed training.

Responsibilities

• Design, train, and deploy large deep learning models leveraging data from a fleet of vehicles.
• Improve training infrastructure and deployment pipelines to accelerate iteration and performance.
• Provide thorough code reviews enforcing ML engineering standards and security.

Skills

PyTorch

CUDA

Triton

Distributed training

Transformer models

Model training

Profiling

Education

PhD in CS/CE/EE

Tools

CUDA

Triton

Job description

Rivian is seeking a full-time Senior ML, AI Infra Engineer to design and operate state-of-the-art infrastructure for training large foundation models and accelerating inference for autonomous driving applications. You will collaborate with software engineers, ML engineers and researchers to push the boundaries of ML systems at scale.

The role emphasizes building robust training pipelines, optimizing performance, and deploying models across a growing fleet of vehicles.