Cuspai

Platform Engineer

London, UK, Remote. Posted 22 days ago

The role

You will build and maintain the ML infrastructure platform to empower AI researchers and materials scientists. You will design cloud-native platforms on Kubernetes and manage the full MLOps stack.

What you will do

  • Design and implement a cloud-native platform on Kubernetes
  • Implement and maintain CI/CD pipelines, model registries, experiment tracking systems, and deployment automation
  • Build infrastructure to support distributed training across many GPUs
  • Ensure 99.9% uptime for the ML platform through monitoring, alerting, and automated recovery systems
  • Use Kapitan to manage complex multi-environment Kubernetes configurations
  • Implement resource management strategies to optimise cloud spending
  • Create tools and abstractions for researchers to go from experiment to production
  • Partner with ML Researchers, Chemists, Materials Scientists and Software Engineers

Required qualifications

  • Deep expertise in building and maintaining ML infrastructure
  • Deep experience with ML workloads on Kubernetes and multi-cloud platforms (AWS, GCP, neoclouds)
  • Strong Python and/or Go programming skills for infrastructure automation
  • Expert-level proficiency in Infrastructure as Code (IaC) using Terraform, Helm, Kapitan, and GitOps workflows
  • Experience operating distributed ML platforms for JAX and/or PyTorch
  • GPU infrastructure experience, specifically managing GPU clusters and performance optimisation
  • Previous experience as a Software Engineer before moving into DevOps / MLOps / ML Infrastructure, or writing code regularly

Nice to have

  • Experience using Flyte, Kapitan, Pants or similar tools
  • Experience with HPC environments and job schedulers (Slurm, PBS)
  • Knowledge of ML serving infrastructure (Triton, TorchServe, KServe)
  • Familiarity with scientific computing workflows and data management
  • Experience with multi-tenant ML platforms
  • Background in supporting research teams

Benefits

  • Competitive salary
  • Equity in CuspAI
  • 28 days holiday (DE, NL, UK) or 21 days holiday (JP, SG, US), in addition to local public holidays
  • 26 weeks parental leave for primary caregiver at full pay
  • 12 weeks parental leave for secondary caregiver at full pay
  • Professional development budget

About the company

CuspAI is a frontier AI company on a mission to solve the breakthrough materials needed to power human progress. The founding team is comprised of researchers in AI, chemistry and engineering.