Remote AI Research Engineer: Scalable LLM/VLM Training

Harnham; Remote; GBP 170,000 - 200,000; Full time

Job summary

A leading deep learning company in the UK seeks an AI Engineer to build and optimise large-scale models. This remote role involves training LLMs/VLMs from scratch and improving model architectures. Candidates should have experience in GPU training and frameworks like PyTorch and CUDA. The opportunity offers a competitive salary of up to £200k with equity options. Join a talented team working on cutting-edge AI without geographical constraints.

Qualifications

• Experience in training large-scale LLMs and VLMs.
• Proficiency in distributed GPU systems and optimisations.
• Strong background in CUDA/Triton kernel development.

Responsibilities

• Train LLMs / VLMs from scratch using distributed frameworks.
• Build and optimise multimodal training pipelines.
• Develop and refine Mixture-of-Experts architectures.
• Write and optimise CUDA / Triton kernels.
• Improve training stability, speed, and memory efficiency.
• Experiment with new architectures and scaling laws.

Skills

GPU training

CUDA

PyTorch

Triton

DeepSpeed

multimodal architectures

Job description

A leading deep learning company in the UK seeks an AI Engineer to build and optimise large-scale models. This remote role involves training LLMs/VLMs from scratch and improving model architectures. Candidates should have experience in GPU training and frameworks like PyTorch and CUDA. The opportunity offers a competitive salary of up to £200k with equity options. Join a talented team working on cutting-edge AI without geographical constraints.