Frontier AI LLM Systems Engineer - London

isomorphiclabs; Greater London; On-site; GBP 90,000 - 150,000; Full time

Job summary

Isomorphic Labs in the United Kingdom is seeking a senior machine learning engineer to advance frontier AI for drug design. You will work on scale training of large language models and collaborate with scientists and engineers to deploy production-ready systems.

You will optimize performance, explore low-precision methods, and contribute to model performance and scaling across interdisciplinary teams, helping accelerate discoveries and new medicines.

Qualifications

• Experience with large-scale distributed training of LLMs.
• Proficiency with JAX or PyTorch.
• Knowledge of parallelism strategies and NCCL.
• Excellent collaboration skills.

Responsibilities

• Implement and optimize LLM post-training methods at scale on frontier models.
• Collaborate with research teams to translate new methods into production-ready systems.
• Prioritize and execute on performance optimization opportunities.
• Evaluate and deploy frameworks for supervised fine-tuning, RL and LLM evaluation.
• Diagnose and fix performance bottlenecks and communication overhead in distributed training and inference systems.
• Deploy low-precision methods to balance performance with accuracy in drug design programs.

Skills

Distributed training

LLMs

Deep learning (JAX/PyTorch)

Parallelism & NCCL

GPU architectures

Collaboration

Tools

XLA

Triton

CUDA

Job description

Isomorphic Labs in the United Kingdom is seeking a senior machine learning engineer to advance frontier AI for drug design. You will work on scale training of large language models and collaborate with scientists and engineers to deploy production-ready systems.

You will optimize performance, explore low-precision methods, and contribute to model performance and scaling across interdisciplinary teams, helping accelerate discoveries and new medicines.