Software Engineer (Scheduler)

AI-first drug design and development company building a unified drug-design engine to advance new medicines.

Series BRecently funded
London, United Kingdom
About Isomorphic Labs

Isomorphic Labs uses predictive and generative AI models, computational biology, and drug-design expertise to develop therapeutics across partnered and internal programs.

View jobs by Isomorphic Labs

Skills

Candidate Availability

Required and preferred rules are kept separate and reflect the wording in the original posting.

About the Role

You will join a newly formed Scheduling team to own and extend the systems that allocate compute to complex machine learning workloads on Kubernetes. You will design scheduling systems for large-scale, topology-sensitive workloads, drive policies like gang scheduling and preemption, work with ML researchers, and build reliable custom Kubernetes controllers, operators, and CRDs.

Requirements

  • Practical experience building or operating cluster schedulers, batch systems, or queuing frameworks (e.g., Kueue, Volcano, Slurm)
  • Deep hands-on Kubernetes experience including custom controllers, operators, and strong familiarity with the K8s scheduler
  • Strong cloud infrastructure experience: compute provisioning, networking, Infrastructure as Code
  • Solid foundation in DevOps/SRE best practices
  • High proficiency in Python or Go

Responsibilities

  • Design, operate, and extend ML workload scheduling systems for batch inference and long-running training
  • Implement scheduling policies such as gang scheduling, topology-aware placement, and preemption
  • Collaborate with ML researchers and the Infrastructure team to troubleshoot bottlenecks and turn research requirements into platform capabilities
  • Build reliable systems with clear failure semantics, runbooks, and SLOs
  • Build and maintain custom Kubernetes controllers, operators, and CRDs

Benefits

  • Hybrid working model