LLM Training Engineer — Distributed Systems (Remote)

poolside; Remote; GBP 80,000 - 100,000; Full time

Job summary

A leading AI company in the UK is looking for a skilled engineer to join their pre-training team focused on Large Language Models. The successful candidate will engage in both programming and implementing architectures and will benefit from a fully remote work environment, flexible hours, and a strong commitment to quality. This role offers the opportunity to work with cutting-edge technology in a diverse and inclusive culture.

Benefits

Fully remote work & flexible hours

37 days/year of vacation & holidays

Health insurance allowance for you and dependents

Company-provided equipment

Wellbeing and home office allowances

Frequent team get-togethers

Diverse & inclusive culture

Qualifications

• Deep knowledge of Transformers is a must.
• Trained LLMs and coded LLMs from scratch.
• Experience in writing high-quality Python, Cython, C/C++, Triton, CUDA code.

Responsibilities

• Follow the latest research on LLMs and source code generation.
• Write high-quality code while babysitting and analyzing experiments.
• Work in the team and stay in touch regarding future steps.

Skills

Experience with Large Language Models (LLM)

Deep knowledge of Transformers

Knowledge/Experience with cutting-edge training tricks

Knowledge/Experience of distributed training

Strong machine learning and engineering background

Programming experience in Python with PyTorch or Jax

C/C++, CUDA, Triton knowledge

Strong algorithmic skills

Linux experience

Job description

A leading AI company in the UK is looking for a skilled engineer to join their pre-training team focused on Large Language Models. The successful candidate will engage in both programming and implementing architectures and will benefit from a fully remote work environment, flexible hours, and a strong commitment to quality. This role offers the opportunity to work with cutting-edge technology in a diverse and inclusive culture.