Software Engineer - AI/ML in 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. A future enabled and enriched by the incredible power of machine learning. 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. About IsoIsomorphic 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 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.Your impact We are building the largest foundation models in biotech and applying them immediately to cure disease. You will play a key role and work at a grand scale to deliver the foundations that make this happen. Joining the Compute Infrastructure team, you will ensure our planet-scale accelerator fleet operates at peak health and efficiency. By partnering with in-house machine learning platforms, performance & scaling team, and AI researchers, you will design the observability systems and automated recovery mechanisms that maximize scientific throughput on every paid GPU-hour. Design, deploy, and scale robust observability systems and telemetry pipelines to monitor fleetwide compute efficiency, hardware health, and workload goodput across distributed clustersDrive hardware efficiency and node reliability across our accelerator fleet, and integrating new hardware to leverage advancementsIdentify compute waste and efficiency bottlenecks across the fleet, partnering with ML and platform teams to actively optimize accelerator utilization and improve workload goodputCollaborate with teams in the AI org, and work closely with the ML Infrastructure team to identify, instrument, and improve canonical efficiency metrics across both training and inference runsContribute to the efforts for consistently improving the reliability of our ML runs. Possess real world experience operating, monitoring, and debugging infrastructure for large-scale AI/ML workloadsHave experience working in cloud compute infrastructure design, preferably GCPPossess strong programmings skillsHave significant experience working and deploying in Kubernetes at scaleFamiliarity with the Nvidia GPU generationsProven track record of building production observability and telemetry stacksNice to have: Have a background in either ML SWE or infrastructure SRE work to build onConceptual understanding of ML workload efficiency paradigmsHave experience leading and delivering projects to multidisciplinary stakeholdersFamiliarity with Google TPU generationsFamiliarity with: workload scheduling; machine learning efficiency research; familiarity with ML-driven R&D cycles; These values help to guide our work and will continue to strengthen it. TogetherTogether 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 companyWe 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 workingIt'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). 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