Your browser does not support javascript! Please enable it, otherwise web will not work for you.

Research Engineer, Infrastructure

Home > Python programming jobs

Research Engineer, Infrastructure in UK new

  • London

Responsibilities

  • Scale the post-training stack for supervised fine-tuning and reinforcement learning on open-weight models.
  • Build distributed rollout infrastructure for model inference, tool execution, and training-experience collection.
  • Create reproducible agent environments for model interaction with documents, tools, and external systems.
  • Own training-run performance and reliability, including GPU utilization, checkpointing, failure recovery, and observability.
  • Identify and remove training and inference bottlenecks.
  • Build internal systems for training-data management and versioning, evaluation, result browsing, and training-job and experiment tracking.
  • Integrate models into production and manage serving, latency, cost, and online evaluation.
  • Partner with researchers to turn experimental workflows into dependable systems while preserving rapid experimentation.

Requirements

  • Experience building ML infrastructure, distributed systems, or research tooling with ownership of performance, reliability, and operation.
  • Strong Python and generalist software engineering skills across backend services, data systems, and internal tools.
  • Familiarity with PyTorch, JAX, or similar ML frameworks.
  • Experience with distributed training or inference, including diagnosing performance bottlenecks and failures across the stack.
  • Understanding of post-training workflows.
  • Experience with reinforcement-learning rollouts, agent environments, or sandboxed tool execution is preferred.
  • Practical judgment about what to build versus adopt and the ability to partner with researchers and ship quickly as needs evolve.

Salary: $230k - $340k/yr

Harvey

Harvey is domain-specific AI for legal and professional services. Adopted by Fortune 500 companies like AT&T, Verizon, Cox, Koch, KKR, Bridgewater, more than 100,000 lawyers across 2,400+ customers in 70 countries and over 75% of AmLaw 100 law firms rely on Harvey to advance legal expertise faste...

Similar positions

Machine Learning Research Engineer

  • Apple
  • Full time
  • USA
  • 10/31/2026
  • Seattle, WA

Security Engineer, Offensive Security

  • Fluidstack
  • Full time
  • USA
  • 10/31/2026
  • Salary: $224k - $275k/yr
  • Remote

Staff Software Engineer, Backend (Observability)

  • Affirm
  • Full time
  • Canada
  • 10/31/2026
  • Salary: $181k - $241k/yr
  • Remote

Staff Software Engineer, Sensing Systems

  • CHAOS Industries
  • Full time
  • USA
  • 10/31/2026
  • Salary: $211k - $290k/yr
  • El Segundo, CA

Machine Learning Engineer, Perception

  • Path Robotics
  • Full time
  • USA
  • 10/31/2026
  • Remote