Senior ML Engineer: AI Infrastructure for Autonomy

Rivian; Greater London; On-site; GBP 70,000 - 90,000; Full time

Job summary

A leading electric vehicle manufacturer in the UK is looking for a full-time Machine Learning Engineer. The successful candidate will work on cutting-edge AI infrastructure aimed at developing state-of-the-art autonomous driving solutions. This role includes designing and deploying large deep learning models, enhancing the training pipeline, and requires a strong background in PyTorch. Applicants should have a PhD or equivalent experience and deep knowledge of AI technologies.

Qualifications

• PhD or equivalent experience in industry.
• Deep understanding of PyTorch and Cuda.
• Experience with distributed training of models.

Responsibilities

• Design, train, and deploy large deep learning models.
• Improve training infrastructure and deployment pipeline.

Skills

Deep knowledge of PyTorch

Experience with Cuda or Triton

Knowledge of model training framework

In-depth knowledge of transformer architecture

Experience of performing large scale distributed training

Profiling model to improve training speed

Education

PhD in CS/CE/EE or equivalent industry experience

Tools

Nvidia TensorRT

Transformers

Job description

A leading electric vehicle manufacturer in the UK is looking for a full-time Machine Learning Engineer. The successful candidate will work on cutting-edge AI infrastructure aimed at developing state-of-the-art autonomous driving solutions. This role includes designing and deploying large deep learning models, enhancing the training pipeline, and requires a strong background in PyTorch. Applicants should have a PhD or equivalent experience and deep knowledge of AI technologies.