Bio ML Staff Engineer — Protein RL & Data Infrastructure

Ai; Greater London; On-site; GBP 110,000 - 180,000; Full time

Benefits offered by this job

30 days annual leave

Pension with 10% employer contribution

Bupa private health cover

Job summary

Ai is a stealth AI-biology startup hiring to build infrastructure for data generation. You’ll own biology ML work, integrating protein models and building RL environments for LLM agents, collaborating with scientists and engineers.

You’ll fine-tune models on experimental data, run scaling studies, and shape what data to measure next. London or San Francisco base, with equity-ready opportunities and open decision culture.

Qualifications

• Hands-on ML for biology with wet-lab data validation.
• Experience with protein language/structure/affinity models (ESM C, Boltz-2, OpenFold3, AlphaFold 3).
• Fine-tuning/post-training pretrained biological models on experimental data.

Responsibilities

• Build RL environments and harnesses for LLM agents using ML models.
• Integrate public protein models and test when predictions are trustworthy.
• Run data scaling and ablations to guide measurements and model improvements.
• Collaborate with scientists, bioinformaticians, and infrastructure engineers.
• Document modelling choices and results for ML and scientific audiences.

Skills

Python

PyTorch

JAX

Reinforcement Learning

Biology ML understanding

Tools

ESM C

Boltz‑2

OpenFold3

AlphaFold 3

Job description

Ai is a stealth AI-biology startup hiring to build infrastructure for data generation. You’ll own biology ML work, integrating protein models and building RL environments for LLM agents, collaborating with scientists and engineers.

You’ll fine-tune models on experimental data, run scaling studies, and shape what data to measure next. London or San Francisco base, with equity-ready opportunities and open decision culture.