Remote

Full-time computer-and-mathematical

We are sharing a specialised full-time consulting opportunity for US-based MLOps and ML systems engineers with production experience in JAX, PyTorch, distributed training infrastructure, and custom GPU kernel development using Pallas or Triton.

This role supports a high-impact generative AI initiative focused on developing and evaluating advanced ML infrastructure tasks for frontier model training. Selected engineers will design technically challenging problems, produce rigorous solutions, assess model-generated outputs, and help establish evaluation standards across training pipelines, distributed systems, framework-level optimisation, and GPU kernel performance.

Key Responsibilities

ML Infrastructure & Training Systems

• Analyse and improve machine learning training infrastructure, deployment workflows, and model-development systems
• Guide research and engineering teams on MLOps, distribut...