Research Engineer (LLM Performance), London

Highlights

• Scale foundational AI models set to transform the biopharmaceutical world.
• Optimize LLM post-training at scale to power real-world drug design programs.
• Apply frontier AI to unlock faster breakthroughs and life-changing medicines.

Ideal candidate

A collaborative performance-minded engineer with large-scale LLM experience who thrives in creative, iterative research environments and cares about applying AI to cure disease.

Requirements

• Significant experience with large-scale distributed training of LLMs
• Experience with deep learning ML frameworks (JAX or PyTorch)
• Knowledge of parallelism strategies and collective communication libraries (e.g. NCCL)
• Good understanding of GPU architectures and performance reasoning
• Excellent collaboration skills