Research Engineer (LLM Performance), London

Sourced from jobera.com

Isomorphic Labs is seeking a Senior Research Engineer to join the model performance and scaling team in London. The role focuses on optimizing LLM post-training methods, distributed training systems, and inference performance for frontier models used in drug discovery. You will collaborate with scientists to translate research into production-ready systems while leveraging deep learning frameworks like JAX or PyTorch.

• Implement and optimize LLM post-training methods at scale on frontier models
• Diagnose and fix performance bottlenecks in distributed training and inference systems
• Deploy low-precision methods to balance performance with accuracy
• Collaborate with research teams to translate new methods into production-ready systems
• Evaluate and deploy frameworks for supervised fine-tuning and reinforcement learning

• 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
• Strong reasoning about performance concepts