Staff ML Training & Inference Infrastructure Engineer

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

Job summary

A reputable electric vehicle manufacturer in London is looking for a skilled Deep Learning Engineer. This role involves optimizing training and inference workflows of large deep learning models on NVIDIA GPU systems. Ideal candidates should hold a PhD in a relevant field and have deep knowledge of PyTorch, as well as experience in large-scale distributed model training. The company values an adventurous spirit and offers an innovative work environment.

Qualifications

• Ph.D. in CS/CE/EE required along with equivalent experience.
• Deep knowledge of PyTorch and model training frameworks necessary.
• Experience in large-scale distributed model training essential.

Responsibilities

• Optimize performance of Deep Learning training workload on NVIDIA GPU systems.
• Design, train, and deploy large deep learning models efficiently.
• Improve latency of model inference during operations.

Skills

Deep knowledge of PyTorch

Knowledge of model training frameworks

In-depth knowledge of transformer architecture

Experience with large scale distributed training

Profiling models and improving speed

Education

PhD in CS/CE/EE or equivalent industry experience

Job description

A reputable electric vehicle manufacturer in London is looking for a skilled Deep Learning Engineer. This role involves optimizing training and inference workflows of large deep learning models on NVIDIA GPU systems. Ideal candidates should hold a PhD in a relevant field and have deep knowledge of PyTorch, as well as experience in large-scale distributed model training. The company values an adventurous spirit and offers an innovative work environment.