Senior Machine Learning, AI Infrastructure,… at Rivian · London

Role Summary

We are looking for a full-time Senior ML, AI Infra Engineer, with deep knowledge and strong

enthusiasm towards establishing a state-of-art AI infrastructure for training very large

foundation models and accelerating model training/inference.

Our mission is to solve the autonomous driving problem. You will work with a team of talented

software engineers, machine learning engineers and research scientists to push the boundary

of state-of-art machine learning models which will enable the next-generation E2E solution of

autonomous drivingWe are looking for a full-time Senior ML, AI Infra Engineer, with deep knowledge and strongenthusiasm towards establishing a state-of-art AI infrastructure for training very largefoundation models and accelerating model training/inference.Our mission is to solve the autonomous driving problem. You will work with a team of talentedsoftware engineers, machine learning engineers and research scientists to push the boundaryof state-of-art machine learning models which will enable the next-generation E2E solution ofautonomous driving

Responsibilities

● Design, train, and deploy large deep learning models that can leverage the vast amount oflabeled and unlabeled data from a fleet of million vehicles● Improve ecosystem for training infrastructure and deployment pipeline to accelerate modeliteration and improve performance.● Provide thorough, constructive code reviews that enforce ML engineering standards, securitypractices, and performance optimization

Qualifications

● PhD in CS/CE/EE, or equivalent, in industry experience● Deep knowledge of PyTorch● Experience with CUDA or Triton language for writing custom ops● Knowledge of model training framework● In-depth knowledge of transformer architecture and ways to accelerate the training and inferenceof transformer models● Experience of performing large scale distributed training of models● A track record of profiling model and doing detective work to improve model training andinference spee