← All roles

Machine Learning Research Engineer

relationrx· Engineering
Apply on company site
LondonFull-timePosted 2mo ago

About the role

Machine Learning Research Engineer at Relation building and scaling generative and predictive models of cellular behaviour, bridging theory and production-ready ML systems in a multi-omics biology setting.

Responsibilities

  • Implement and optimise large models, partnering with ML Scientists to translate research ideas into reproducible, scalable training pipelines.
  • Profile and optimise training across compute, memory, and I/O, pursuing measurable gains in throughput, convergence, and stability.
  • Design and implement distributed training strategies across multi-GPU and multi-node configurations.
  • Build and maintain core ML infrastructure.
  • Contribute to architectural and algorithmic decisions, bringing engineering judgment into research discussions.
  • Optimise inference and downstream deployment so models can be used by data scientists and biologists in our discovery workflows.
  • Address numerical, performance, and reliability issues across the stack.
  • Establish and maintain engineering practices in research code.
  • Track developments in ML systems and bring relevant advances into our stack.

Qualifications

  • A degree in Computer Science, Engineering, Physics, or a related quantitative discipline; industry experience as an ML / research engineer working on large neural network training.
  • Strong software engineering fundamentals in Python and deep expertise in PyTorch (or equivalent modern ML frameworks).
  • Hands-on experience training large neural networks at scale, including distributed training frameworks.
  • Demonstrable experience profiling and optimising GPU workloads.
  • Working knowledge of cloud-based ML infrastructure and containerised environments.
  • A track record of taking research code from prototype to robust, reusable infrastructure that other people actually use.

Skills

Required:PythonPyTorchDistributed trainingGPU profiling/optimisationCloud ML infrastructure
Last updated Oct 21, 2026. Interested? Apply on the company site.

About relationrx

Discovering biology’s relationships, curing disease.

Discovering biology’s relationships, curing disease.

biotechmedicinebiologydiseasegenomicstherapy

More open roles at relationrx