Head of Edge AI & Hardware ML Deployment

Rivian; Greater London; On-site; GBP 120,000 - 190,000; Full time

Job summary

Rivian is seeking a senior leader to drive the development of edge AI platforms and ML deployment at scale. You will shape compiler tooling, model ingestion, and quantization-aware training while coordinating across hardware and software teams.

You will build and mentor a world-class team focused on model deployment, compiler infrastructure, and hardware-aware ML, guiding architecture and performance across future Rivian platforms.

Qualifications

• Proven experience leading or managing teams in machine learning systems, model deployment, or AI infrastructure.
• Strong background in model optimization for edge or accelerated inference, including quantization and performance tuning.
• Experience building or extending compiler, toolchain, or graph-transformation infrastructure for ML workloads.
• Strong understanding of hardware/software co-design for inference acceleration.
• Experience working with low-power edge platforms, NPUs, custom accelerators, GPUs, or closely related architectures.
• Ability to reason about memory allocation, execution scheduling, bandwidth constraints, and overall system performance.
• Experience translating research concepts into practical tooling and production workflows.
• Exceptional cross-functional leadership skills and a history of driving impact through technical and strategic influence.

Responsibilities

• Direct the development and strategy of machine learning enablement workflows for Rivian edge AI platforms.
• Oversee the roadmap for front-end compiler and model-ingestion tooling to support an expanding model zoo on RAP1 and future platforms.
• Drive the architectural vision for robust quantization-aware training and model optimization infrastructure at scale.
• Manage the end-to-end process of porting, optimizing, and benchmarking machine learning models for hardware accelerators and NPUs.
• Lead cross-functional initiatives with hardware and systems teams to optimize performance across memory, scheduling, and data-movement constraints.
• Establish rigorous standards and methods for evaluating model fidelity and deployment quality across diverse execution paths.
• Collaborate with executive leadership to identify highest-value workloads and align technical execution with Rivian's product milestones.
• Recruit, mentor, and grow a world-class team of engineers specializing in model deployment, compiler infrastructure, and hardware-aware ML.
• Help shape the long-term direction of Rivian’s edge AI stack, including support for increasingly large and complex neural network models.

Skills

Team leadership

Edge ML

Model optimization

Compiler infra

Hardware-aware ML

Cross-functional collaboration

Tools

ONNX

Graph transformation

ML toolchain

NPUs/GPUs

Job description

Rivian is seeking a senior leader to drive the development of edge AI platforms and ML deployment at scale. You will shape compiler tooling, model ingestion, and quantization-aware training while coordinating across hardware and software teams.

You will build and mentor a world-class team focused on model deployment, compiler infrastructure, and hardware-aware ML, guiding architecture and performance across future Rivian platforms.