In this role you will design and implement improvements to our large scale training infrastructure and directly help make technical decisions to optimise our models' training performance and efficiency. Strong engineering skills with fluency in Python and PyTorch or other acceleration libraries. Experienced in scaling up GPU jobs via large-scale compute clusters using Slurm or Kubernetes.