Founding ML Systems Engineer — Distributed Training Lead

Dex; Greater London; On-site; GBP 120,000 - 180,000; Full time

Job summary

Dex in Greater London is seeking a senior hands-on ML engineer to own the entire ML stack, including model code, training workflows, and experiment infrastructure. You will build distributed training capabilities for large-scale models and integrate research branches into a coherent production-ready codebase.

You will also design the backend platform that delivers foundation models to customers, optimize training performance, and shape the 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, improving system efficiency.
• Design and build the backend platform that delivers foundation models to customers.

Skills

Senior engineer

PyTorch

Distributed training

Model engineering

Python backend

Tools

Docker

Kubernetes

Infrastructure as Code

Job description

Dex in Greater London is seeking a senior hands-on ML engineer to own the entire ML stack, including model code, training workflows, and experiment infrastructure. You will build distributed training capabilities for large-scale models and integrate research branches into a coherent production-ready codebase.

You will also design the backend platform that delivers foundation models to customers, optimize training performance, and shape the engineering culture from day one.