← All roles

Research Engineer (LLM Performance), London

Isomorphic Labs· ML Research
Apply on company site
Hybrid (London)Full-timePosted 1mo ago

About the role

Research Engineer focused on scaling and optimizing large language models (LLMs) for frontier AI applications in drug discovery, joining the model performance and scaling team to advance production-ready systems and performance in biopharma contexts.

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.
  • Relentlessly prioritize and execute on performance optimization opportunities.
  • Evaluate and deploy frameworks for supervised fine-tuning, reinforcement learning 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, impacting real world drug design programs.

Qualifications

  • Significant experience with large scale distributed training of LLMs.
  • Experience with deep learning ML frameworks (either JAX or PyTorch).
  • Knowledge of parallelism strategies and collective communication libraries (e.g. NCCL).
  • Good understanding of GPU architectures. Reasoning about performance concepts is more important than writing kernels from scratch
  • Excellent collaboration skills.

Skills

Required:Large-scale distributed training of LLMsDeep learning frameworks (JAX or PyTorch)Parallelism and collective communication (NCCL)GPU architectures
Last updated Sep 9, 2026. Interested? Apply on the company site.

About Isomorphic Labs

Reimagining Drug Discovery Process with AI

Isomorphic Labs is building a future where frontier AI can help to unlock deeper scientific insights, faster breakthroughs, and life-changing medicines.

AIBiotechnologyDrug DiscoveryPharmaceuticalsliquid-ai-recruiting-target
Company profileWebsite

More open roles at Isomorphic Labs