Impress employers and recruiters.
Choose from hundreds of resume examples.
Impress employers and recruiters.
Choose from hundreds of resume examples.
Tailor your resume to this Research Engineer (LLM Performance), London role.
Rezi rewrites your resume against Isomorphic Labs's job description. Free.
Tailor your resume to this Research Engineer (LLM Performance), London role.
Rezi rewrites your resume against Isomorphic Labs's job description. Free.
Don't guess if your resume is good enough.
See how it scores against the Research Engineer (LLM Performance), London posting at Isomorphic Labs — free, in seconds.
Don't guess if your resume is good enough.
See how it scores against the Research Engineer (LLM Performance), London posting at Isomorphic Labs — free, in seconds.
About the Role
This role offers an exciting opportunity to contribute to frontier research at the intersection of AI and drug design. You will join the model performance and scaling team, partnering with scientists and engineers to scale foundational models that will transform the biopharmaceutical world. You will apply 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.
- 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.
Requirements
- 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
- JAX
- PyTorch
- NCCL
- GCP
- XLA
- Triton
- Pallas
- CUDA
Location
- Hybrid
Work Type
- Hybrid
- Full-time
Experience Level
- Senior
About the Company
- Isomorphic Labs is applying frontier AI to help unlock deeper scientific insights, faster breakthroughs, and life-changing medicines with an ambition to solve all disease.
- Isomorphic Labs (IsoLabs) was launched in 2021 to advance human health by building on and beyond the Nobel-winning AlphaFold system.
- Our interdisciplinary team of drug discovery experts and machine learning specialists has built powerful new predictive and generative AI models that accelerate scientific discovery at digital speed.
- Our name comes from the belief that there is an underlying symmetry between biology and information science.
- By harnessing AI’s powerful capabilities, we can use it to model complex biological phenomena to help design novel molecules, anticipate how drugs will perform and develop innovative medicines to treat and cure some of the world’s most devastating diseases.
- We have built a world-leading drug design engine comprising AI models that are capable of working across multiple therapeutic areas and drug modalities.
- We are continually innovating on model architecture and developing cutting-edge capabilities to advance rational drug design.
- Every day, and with each new breakthrough, we’re getting closer to the promise of digital biology, and achieving our ambitious mission to one day solve all disease with the help of AI.
Equal Opportunity
- We are committed to equal employment opportunities regardless of sex, race, religion or belief, ethnic or national origin, disability, age, citizenship, marital, domestic or civil partnership status, sexual orientation, gender identity, pregnancy or related condition (including breastfeeding) or any other basis protected by applicable law.
- If you have a disability or additional need that requires accommodation, please do not hesitate to let us know.