Staff ML Infra Engineer: Self-Driving Training & Inference

Rivian; Greater London; On-site; GBP 80,000 - 100,000; Full time

Job summary

A leading electric vehicle manufacturer in Greater London seeks a Staff Software Engineer specializing in machine learning infrastructure. The ideal candidate will optimize deep learning performance on NVIDIA GPU systems and manage large-scale model training. Applicants should possess a PhD in Computer Science or equivalent experience, and have substantial knowledge in PyTorch and deep learning frameworks. This role contributes directly to safety-critical self-driving technologies for innovative vehicles.

Qualifications

• Deep knowledge of machine learning infrastructure for autonomous driving.
• Experience optimizing deep learning performance on NVIDIA GPU systems.
• Proven track record of improving model training and inference speeds.

Responsibilities

• Optimize performance of Deep Learning training workload on large-scale systems.
• Optimize latency of model inference on onboard systems.
• Design, train, and deploy large deep learning models.

Skills

Deep knowledge of PyTorch

Knowledge of model training frameworks

In-depth knowledge of transformer architecture

Experience with large scale distributed training of models

Performance optimization for model training

Education

PhD in CS/CE/EE or equivalent industry experience

Job description

A leading electric vehicle manufacturer in Greater London seeks a Staff Software Engineer specializing in machine learning infrastructure. The ideal candidate will optimize deep learning performance on NVIDIA GPU systems and manage large-scale model training. Applicants should possess a PhD in Computer Science or equivalent experience, and have substantial knowledge in PyTorch and deep learning frameworks. This role contributes directly to safety-critical self-driving technologies for innovative vehicles.