Machine Learning Performance Engineer, London London 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. The future is coming. A future enabled and enriched by the incredible power of machine learning. A future in which diseases are curtailed or cured starting with better and faster drug discovery. 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. The world we want tomorrow is the one were building today. It starts with the culture at this company. It starts with you. About Iso 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. Our name comes from the belief that there is an underlying symmetry between biology and information science. By harnessing AIs 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 worlds 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, were getting closer to the promise of digital biology, and achieving our ambitious mission to one day solve all disease with the help of AI. 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. You will draw upon your existing engineering experience whilst learning from those around you, to apply novel techniques and ideas to newly encountered model and systems performance optimizations, as well as machine learning, computational biology and chemistry problems.

Make sure to read the full description below, and please apply immediately if you are confident you meet all the requirements.

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 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. Experience with deep learning ML frameworks (preferably JAX). 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 its also about initiative and integrity. The scale of the challenge demands nothing less. Determined Determined at Iso is the way we pursue our goal. Its a confidence in our hypothesis, as well as the urgency and agility needed to deliver on it. Because disease wont wait, so neither should we. Together Together at Iso is about connection, collaboration across fields and catalytic relationships. Its knowing that transformation is a group project, and remembering that what were 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. Its 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 youre in). If you have additional needs that would prevent you from following this hybrid approach, wed be happy to talk through these if youre selected for an initial screening call. Please note that when you submit an application, your data will be processed in line with our privacy policy . Interested in building your career at Isomorphic Labs? Get future opportunities sent straight to your email. Apply for this job indicates a required field First Name * Last Name * Email * Phone Resume/CV * Enter manually Accepted file types: pdf, doc, docx, txt, rtf As an AI-first company, we embrace tools like Gemini. However, our interview process is designed to understand your unique expertise and problem-solving skills, as well as what motivates you.Therefore, we ask that you refrain from using any AI tools, transcription services, or other assistants during your interview and assessment process, unless you've made prior arrangements with our Talent team for specific needs or accommodations. (If you require any adjustments, please discuss this with the Talent team during your initial screening call.) * By checking this box and proceeding with your application, you confirm that you have read and agreed to these terms and will not use any AI tools or assistants during the interview and assessment process. In which of our office locations would you prefer to be based? (Note that certain jobs may only be available in one location - please check the details before applying) * Select... What are your salary expectations (base / total package)? Please note that this is what you would be looking for in your next role, NOT your current package. What is your current notice period? Where did you hear about Iso? Press article / publication DeepMind website LinkedIn Twitter Indeed Other job board Search engine Conference/event (please let us know which one below) Alphabet Grow (current employee of GDM) Alphabet Grow (current employee of any other Bet) Referred by an employee at Isomorphic Labs (please let us know who below) Other Where did you hear about Iso? (Additional info) Is there anything else you'd like us to know? LinkedIn Profile Website Iso Demographic Questions We value diversity of experience, knowledge, backgrounds and perspectives and harness these qualities to create extraordinary impact. We are working to build teams that reflect and represent the populations we are striving to serve. As part of this effort we would like to better understand our candidate audience, so that we can continue to improve. We need your help to do this, but filling in this form is entirely voluntary . You may choose not to provide the requested demographic information. Whether you choose to provide the information or not, this will be kept separate from your application and will have no bearing on any hiring decision. Any demographic information you provide will be anonymised and held separately from your application . No Isomorphic Labs hiring decision makers will have access to this information and the information will only be used for aggregated reporting and monitoring purposes, not for the purposes of your application or making any hiring decisions. You can find more information about how we pr