Founding ML Systems Engineer — Distributed Training

Dex; Greater London; On-site; GBP 90,000 - 160,000; Full time

Job summary

Dex is seeking a senior, hands-on ML engineer to own the full ML stack for foundation models, from model code to training workflows and backend platform delivery. You will lead distributed training on multi-GPU clusters, integrate research branches into production, and ship robust tooling for experiment tracking and hyperparameter optimisation.

This founding role sits at an early-stage company in Greater London, offering significant impact and the chance to shape engineering culture from day one.

Qualifications

• Senior, hands-on engineer who still writes and ships critical code.
• Strong practical experience with PyTorch across model code, data pipelines, and training loops.
• Proven track record with distributed training (multi-GPU/node, GPU clusters).
• Real model-engineering experience in physics/simulation, vision, or LLM systems.

Responsibilities

• Own the entire ML stack: model code, training and evaluation workflows, and experiment infrastructure.
• Build and implement distributed training capabilities for large-scale model development.
• Integrate independently developed research branches into a coherent, production-ready codebase.
• Profile and resolve bottlenecks in training stability and performance to improve system efficiency.
• Design and build the backend platform that delivers these foundation models to customers.

Skills

PyTorch

Distributed training

Python

GPU clusters

Docker/Kubernetes/IaC

Tools

Docker

Kubernetes

Terraform/IaC

Job description

Dex is seeking a senior, hands-on ML engineer to own the full ML stack for foundation models, from model code to training workflows and backend platform delivery. You will lead distributed training on multi-GPU clusters, integrate research branches into production, and ship robust tooling for experiment tracking and hyperparameter optimisation.

This founding role sits at an early-stage company in Greater London, offering significant impact and the chance to shape engineering culture from day one.