Research Engineer (LLM Performance), London
Salary not listed • London • Full-time
Posted 1mo ago
Job Location
London
Tech Stack
Remote Work Policy
On-site
Employment Type
Full-time
Categories
LLM Engineer
About the job
Isomorphic Labs is applying frontier AI to drug discovery to accelerate scientific breakthroughs and develop life-changing medicines. We are building a world-leading drug design engine using predictive and generative AI models. This role is an exciting opportunity to contribute to frontier research at the intersection of AI and drug design, working in a highly creative, iterative environment within the model performance and scaling team. You will partner with scientists and engineers to scale foundational models that will transform the biopharmaceutical world, applying novel techniques to model and systems performance optimizations, as well as machine learning, computational biology, and chemistry problems.
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, 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 for drug design programs.
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 concepts.
- Excellent collaboration skills.