Isomorphic Labs is hiring a

Software Engineer (ML, Performance Engineering), London

London, United Kingdom

Software Engineer, ML - Performance Engineering

We are looking for engineers with different levels of experience - Mid through to Senior, Staff or equivalent levels. Isomorphic Labs is a new Alphabet company that is reimagining drug discovery through a computational- and AI-first approach. 

We are on a mission to accelerate the speed, increase the efficacy and lower the cost of drug discovery. You'll be working at the cutting edge of the new era of 'digital biology' to deliver a transformative social impact for the benefit of millions of people.

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

Your impact 

Working in a highly creative, iterative environment, you will be partnering with leading engineers, scientists and ML researchers to build the critical platform driving that transformation. This is a newly created role and you will need to use your previous experience and show initiative in order to fully carve out your contribution. This is an exciting opportunity for you to shape the new performance and efficiency focused capability of Isomorphic Labs. 


What you will do

  • Develop highly efficient GPU kernels maximising utilisation and application performance.
  • Implement aggressive optimisations to overcome hardware limitations.
  • Design low-precision algorithms balancing high performance with minimal accuracy loss.
  • Work with ML engineers to create training and inference-friendly model architectures.
  • Use your intuitions of both runtime and hardware on-device profiling to remove model inference as a bottleneck in quick drug design iteration cycles.
  • Partner with scaling teams to deploy kernels and manage training uptime.
  • Profile jax level code, and implement improvements.
  • Shorten the drug design iteration cycles by adapting models to inference heavy workloads.

Skills and qualifications

Essential

  • Deep understanding of GPU and other AI accelerator architectures.
  • Excellent collaboration skills with infrastructure and ML engineers.
  • Experience with modern ML frameworks in production settings
  • 3+ years of relevant industry experience.
  • Strong understanding of data structures and algorithms.
  • Good understanding of maths.
  • Ability to code up a quick prototype solution in Python. 

Nice to have

  • Experience with XLA, Triton, CUDA, Pallas or similar accelerator programming languages/deep learning compiler stacks
  • Interest in chemistry and biology.
  • Proven ability to optimise ML accuracy with low-precision formats.
  • Experience building, deploying and maintaining production systems on GCP.


Culture and values

What does it take to be successful at IsoLabs? It's not about finding people who think and act in the same way, but we do have some shared values:

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.

In this 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 inclusive company

We realise that to be successful we need our teams to reflect and represent the populations we are striving to serve. We’re working to build a supportive and inclusive environment where collaboration is encouraged and learning is shared. We value diversity of experience, 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 be able to share knowledge and establish relationships with each other, and we find it easier to do this if we spend time together in person. This is why we’ve decided to follow a hybrid model, and would require you to be able to come into the office 3 days a week (currently Tue, Wed, and one other day depending on which team you’re in).  As an equal opportunities employer we are committed to building an equal and inclusive team. 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.


>> Click to view other open roles at Isomorphic Labs

Apply for this job

Please mention you found this job on AI Jobs. It helps us get more startups to hire on our site. Thanks and good luck!

Get hired quicker

Be the first to apply. Receive an email whenever similar jobs are posted.

Ace your job interview

Understand the required skills and qualifications, anticipate the questions you may be asked, and study well-prepared answers using our sample responses.

Performance Engineer Q&A's
Report this job
Apply for this job