Member of Technical Staff, ML Infrastructure

Ai; Greater London; On-site; GBP 90,000 - 150,000; Full time

Job summary

Ai is building the infrastructure for cutting-edge AI and biology research in London or San Francisco. You will own the end-to-end stack for post-training and reinforcement learning, enabling researchers and AI agents to run, monitor and compare experiments with reproducible data.

The role emphasizes strong system design, scalable GPU-based workflows, and collaboration with RL, evaluation, Bio ML researchers and engineers.

Qualifications

• Experience building infrastructure for LLM inference or RL training.
• Systems fundamentals; ability to profile a slow or failing workload to its cause.
• Hands-on with GPU clusters, distributed training, model serving or high-throughput inference; scheduling, utilisation and cost.
• Docker, Kubernetes, Ray or Terraform; long-running jobs with retries, checkpointing and clear failure reporting.
• Good judgement on technical trade-offs, such as when a quick prototype is enough and when a system needs to be hardened, simplified or rebuilt.
• Clear writing and communication about system design, operational risk and engineering trade-offs.
• Fluent Python and daily use of coding agents.

Responsibilities

• Build infrastructure for post-training and reinforcement learning.
• Support training/post-training runs and inference serving; evaluate harnesses.
• Provide clear interfaces for researchers and AI agents to launch, monitor and debug runs.
• Own the stack end to end: design, implementation, deployment and maintenance.
• Set engineering standards: testing, observability and versioning; remove bottlenecks across training, rollout, evaluation and data movement.

Skills

LLM inference infra

RL training infra

GPU clusters

Profiling performance

Python

Docker

Kubernetes

System design

Cost optimization

Tools

Ray

Terraform

Docker

Kubernetes

Job description

THE OPPORTUNITY

We're a stealth startup in AI and bio, building infrastructure for data generation.

The team is small and elite, the problems are hard, and the foundations are being laid right now.

You'd help write them from the first line, with real ownership and a direct line to the founders.

ABOUT US

AI for biology has a data problem.

The data that matters most doesn't exist yet, so we're building the infrastructure to produce it, with quality and traceability built in from day one.

We're in stealth and heads-down on execution.

We'll share more about what we're building once you've spoken with the team.

THE ROLE

You will build the infrastructure for post-training and reinforcement learning, so that the team can post-train models, benchmark them, and build and train agents.

That covers training and post-training runs, inference serving for open and closed models, the systems that run agents through our RL environments, the evaluation harnesses, and the data pipelines that feed them.

The stack is used by researchers and by AI agents alike, so it needs clear interfaces that both can use to launch, monitor and debug runs.

You will own each part of the stack end to end, from design and implementation through deployment, documentation and maintenance.

You will set its engineering standards, including testing, observability and versioning, and find and remove bottlenecks across training, rollout generation, evaluation runs, data movement and GPU utilisation.

Our data comes from our own experiments and continues to grow over time, so every training run and evaluation needs to be reproducible on each new version of the data and comparable with earlier runs.

You will also improve the performance, scalability and reliability of the stack as the team and its workloads grow.

You will report into the intelligence team and work closely with the RL, evaluation and Bio ML researchers and engineers, as well as with the software team responsible for capturing experimental data.

WHO YOU ARE

You have built and operated infrastructure that researchers relied on for training, serving or reinforcement learning.

A background in biology is not required.

The role is based in London or San Francisco.

MUST HAVE

• Experience building infrastructure for LLM inference or RL training.
• Systems fundamentals, and the ability to profile a slow or failing workload down to its cause.
• Hands-on work with GPU clusters, distributed training, model serving or high-throughput inference, including scheduling, utilisation and cost.
• Docker, Kubernetes, Ray or Terraform, and long-running jobs with retries, checkpointing and clear failure reporting.
• Good judgement on technical trade-offs, such as when a quick prototype is enough and when a system needs to be hardened, simplified or rebuilt.
• Clear writing and communication about system design, operational risk and engineering trade-offs.
• Fluent Python and daily use of coding agents.

NICE TO HAVE

• Inference engines such as SGLang or vLLM, and RL training frameworks such as veRL, Miles or TRL.
• Agent environment and evaluation frameworks such as Harbor.
• Sandboxed execution for agents, such as Firecracker or gVisor.
• Compilers, custom kernels or optimisation for specific GPU hardware.
• Data versioning and lineage for ML datasets.
• Experience building infrastructure for biological or other scientific data.

HOW WE WORK

We work in the open, write decisions down, and expect each person to own their area and make the calls inside it.

• Everyone gets 30 days of annual leave plus local public holidays.
• In the UK that comes with a pension with a 10% employer contribution and Bupa private health cover; in the US, a 401(k) with matching and health cover.
• We add to these as the team grows.

OUR PROCESS

Four stages: a screening call, a behavioural interview, an in-person technical session with the team on a real problem from our work, and references.

We are an equal opportunity employer. We make hiring decisions on merit, scope-fit, and the strength of the working relationship we expect to build with each hire.

Applications welcome from candidates of any background.