Machine Learning Performance Engineer - London

Isomorphic Labs
London
2 days ago
Create job alert

Machine Learning Performance Engineer, London

London

Machine Learning Performance Engineer, London

We are here to advance human health, by reimagining drug discovery with the power and pace of artificial intelligence.

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 by better and faster drug discovery.

Our values exist in service of that future. We think they’ll help us bring it closer, too.

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 we’re building today. It starts with the culture at this company. It starts with you.

About Isomorphic Labs

Isomorphic Labs (IsoLabs) was founded in 2021 and is led by Sir Demis Hassabis. Our aim is to usher in a new era of biomedical breakthroughs and find cures for some of humanity’s devastating diseases.

Our foundations are built on the success of Google DeepMind’s AlphaFold, but we didn’t stop there! We are continuing to develop and implement state-of-the-art technologies as we move towards our goal of dramatically accelerating and improving the process of designing and bringing new medicines to patients.

We have built a world-leading drug design engine comprising foundational AI models that are capable of working across multiple therapeutic areas and drug modalities. The company is continually innovating on model architecture and developing cutting-edge capabilities to advance rational drug design.

Your impact

We are looking for engineers with different levels of experience - Mid through to Senior, Staff or equivalent levels.

This is an exciting opportunity for you to shape the new performance and scaling focused capability of Isomorphic Labs.

What you will do

  • Develop custom GPU kernels maximising utilisation and application performance.
  • Help design, implement, and optimize advanced distributed training and inference strategies.
  • Implement low-level optimisations to overcome hardware limitations.
  • Design low-precision algorithms balancing high performance with minimal accuracy loss.
  • Use your intuitions of both runtime and hardware to optimize performance for both latency and throughput on real world drug design programs.
  • Partner with technical infrastructure teams to deploy solutions and manage training uptime.
  • Work with ML engineers and researchers to create efficient training and inference model architectures.

Skills and qualifications

  • 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, CUDA, Pallas or similar accelerator DSLs / compiler stacks.
  • Experience with distributed training and data/model sharding strategies.
  • Knowledge of collective communication libraries (e.g. NCCL).
  • Proven ability to optimise ML accuracy with 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.

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, andwould 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). If you have additional needs that would prevent you from following this hybrid approach, we’d be happy to talk through these if you’re 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 .

Accepted file types: pdf, doc, docx, txt, rtf

LinkedIn Profile

Website

Are you based in, or would you be willing to relocate to the UK or Switzerland? * Select...

What are you ideally looking for in terms of salary expectations (base / total package)? Please note that this is NOT your current package. (Optional)

What is your current notice period?

What would your preferred working approach be in terms of coming into the office (assuming it is safe to do so)? (Optional) Select...

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?

Demographic Questions - UK & CH

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, butfilling 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 beanonymised 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 process diversity and demographic information in our Applicant and Candidate Privacy Notice .

If you have any questions about the form contact us: .

Is the gender you identify with the same as your gender registered at birth? Select...

What is your ethnicity? (Ethnic origin is not about nationality, place of birth or citizenship. It is about the group to which you perceive you belong.) Select...

#J-18808-Ljbffr

Related Jobs

View all jobs

Head of Software Engineering | £150k – Java, Machine Learning, and Data-Driven Innovation

Head of Software Engineering | £150k – Java, Machine Learning, and Data-Driven Innovation

Staff Machine Learning Engineer Sydney

Machine Learning Ops Engineer

Machine Learning Engineer

Machine Learning Ops Engineer

Get the latest insights and jobs direct. Sign up for our newsletter.

By subscribing you agree to our privacy policy and terms of service.

Industry Insights

Discover insightful articles, industry insights, expert tips, and curated resources.

Portfolio Projects That Get You Hired for AI Jobs (With Real GitHub Examples)

In the fast-evolving world of artificial intelligence (AI), an impressive portfolio of projects can act as your passport to landing a sought-after role. Even if you’ve aced interviews in the past, employers in AI and machine learning (ML) are increasingly asking candidates to demonstrate hands-on experience through the projects they’ve built and shared online. This is because practical ability often speaks volumes about your suitability for a role—far more than any exam or certification alone could. In this article, we’ll explore how to build an outstanding AI portfolio that catches the eye of recruiters and hiring managers, including: Why an AI portfolio is crucial for job seekers. How to choose AI projects that align with your target roles. Specific project ideas and real GitHub examples to help you stand out. Best practices for showcasing your work, from writing clear READMEs to using Jupyter notebooks effectively. Tips on structuring your GitHub so that employers can instantly see your value. Moreover, we’ll discuss how you can use your portfolio to connect with top employers in AI, with a handy link to our CV-upload page on Artificial Intelligence Jobs for when you’re ready to apply. By the end, you’ll have a clear roadmap to building a portfolio that will help secure interviews—and the AI job—of your dreams.

AI Job Interview Warm‑Up: 30 Real Coding & System‑Design Questions

In today's competitive AI job market, nailing a technical interview can be the difference between landing your dream role and getting lost in the crowd. Whether you're looking to break into machine learning, deep learning, NLP (Natural Language Processing), or data science, your problem-solving skills and system design expertise are certain to be put to the test. AI‑related job interviews typically involve a range of coding challenges, algorithmic puzzles, and system design questions. You’ll often be asked to delve into the principles of machine learning pipelines, discuss how to optimise large-scale systems, and demonstrate your coding proficiency in languages like Python, C++, or Java. Adequate preparation not only boosts your confidence but also reduces the likelihood of fumbling through unfamiliar territory. If you’re actively seeking positions at major tech companies or innovative AI start-ups, then check out www.artificialintelligencejobs.co.uk for some of the latest vacancies in the UK. Meanwhile, this blog post will guide you through 30 real coding & system-design questions you’re likely to encounter during your AI job interview. This list is designed to help you practise, anticipate typical question patterns, and stay ahead of the competition. By reading through each question and thinking about the possible approaches, you’ll sharpen your problem-solving skills, time management, and critical thinking. Each question covers fundamental concepts that employers regularly test, ensuring you’re well-equipped for success. Let’s dive right in.

Negotiating Your AI Job Offer: Equity, Bonuses & Perks Explained

Artificial intelligence (AI) has proven itself to be one of the most transformative forces in today’s business world. From smart chatbots in customer service to predictive analytics in finance, AI technologies are reshaping how organisations operate and innovate. As the demand for AI professionals grows, so does the complexity of compensation packages. If you’re a mid‑senior AI professional, you’ve likely seen job offers that include far more than just a base salary—think equity, bonuses, and a range of perks designed to entice you into joining or staying with a company. For many, the focus remains squarely on salary. While that’s understandable—after all, your monthly take‑home pay is what covers day-to-day expenses—limiting your negotiations to salary alone can leave considerable value on the table. From stock options in ambitious startups to sign‑on bonuses that ‘buy you out’ of your current contract, modern AI job offers often include elements that can significantly boost your long-term wealth and job satisfaction. This article aims to shed light on the full scope of AI compensation—specifically focusing on how equity, bonuses, and perks can enhance (or sometimes detract from) the overall value of your package. We’ll delve into how these elements work in practice, what to watch out for, and how to navigate the negotiation process effectively. Our goal is to provide mid‑senior AI professionals with the insights and tools to land a holistic compensation deal that accurately reflects their technical expertise, leadership potential, and strategic importance in this fast-moving field. Whether you’re eyeing a leadership role in machine learning at an established tech giant, or you’re considering a pioneering position at a disruptive AI startup, the knowledge in this guide will help you weigh the merits of base salary alongside the potential riches—and risks—of equity, bonuses, and other benefits. By the end, you’ll have a clearer sense of how to align your compensation with both your immediate lifestyle needs and long-term career aspirations.