Machine Learning Performance Engineer, London

Join to apply for the Machine Learning Performance Engineer, London role at myGwork - LGBTQ+ Business Community

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.

Come and be part of an interdisciplinary team driving groundbreaking innovation and play a meaningful role in contributing towards us achieving our ambitious goals, while being a part of an inspiring and collaborative culture.

About Isomorphic Labs

Isomorphic Labs (IsoLabs) was launched in 2021 to advance human health by building on and beyond the Nobel-winning AlphaFold system. Since then, 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.

Machine Learning Performance Engineer, London Your impact

This is an exciting opportunity for you to contribute to frontier research at the intersection of AI and drug design.

Working in a highly creative, iterative environment, you will join the model performance and scaling team, where you will partner with scientists and engineers to scale foundational models that will transform the biopharmaceutical world as we know it.

What you will do

• Develop custom GPU kernels maximising utilisation and application performance.
• Help design, implement, and optimize distributed training and inference strategies.
• Implement low-level optimisations to overcome hardware limitations.
• Diagnose and debug performance bottlenecks to optimize latency and throughput, impacting real world drug design programs.
• Design low-precision algorithms balancing high performance with minimal accuracy loss.
• Collaborate with researchers to create efficient training and inference model architectures.

Skills And Qualifications Essential

• Significant industry experience and strong working knowledge of HPC and ML systems.
• Good understanding of GPU and other AI accelerator architectures.
• Strong knowledge of data structures and algorithms.
• Good working knowledge of maths / linear algebra.
• Experience with deep learning ML frameworks (preferably JAX).
• Excellent collaboration skills.

Nice To Have

• Knowledge of XLA, Triton, Pallas, CUDA or similar accelerator DSLs / compiler stacks.
• Experience with distributed training and data/model sharding strategies.
• Knowledge of collective communication libraries (e.g. NCCL).
• Experience with optimising ML accuracy using low-precision formats.
• Experience building, deploying and maintaining production systems on GCP.
• Interest in chemistry and biology.

Culture and values

We are guided by our shared values. It's not about finding people who think and act in the same way. These values help to guide our work and will continue to strengthen it.

Thoughtful

Thoughtful at Iso is about curiosity, creativity and care. It is about good people doing good, rigorous and future-making science every single day.

Brave

Brave at Iso is about fearlessness, but it’s also about initiative and integrity. The scale of the challenge demands nothing less.

Determined

Determined at Iso is the way we pursue our goal. It’s a confidence in our hypothesis, as well as the urgency and agility needed to deliver on it. Because disease won’t wait, so neither should we.

Together

Together at Iso is about connection, collaboration across fields and catalytic relationships. It’s knowing that transformation is a group project, and remembering that what we’re doing will have a real impact on real people everywhere.

Creating An Extraordinary Company

We believe that to be successful we need a team with a range of skills and talents. We're building an environment where collaboration is fundamental, learning is shared and every employee feels supported and able to thrive. We value unique experiences, knowledge, backgrounds, and perspectives, and harness these qualities to create extraordinary impact.

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.

Hybrid working

It’s hugely important for us to share knowledge and build strong relationships with each other, and we find it easier to do this if we spend time together in person. This is why we follow a hybrid model, and would require you to be able to come into the office 3 days a week (currently Tuesday, Wednesday, and one other day depending on which team you’re in).