Head of ML Systems & Production

Digital Waffle; United Kingdom; On-site; GBP 120,000 - 180,000; Full time

Job summary

Digital Waffle is building a scalable ML infrastructure and seeks a hands-on leader to own the full ML lifecycle, from data to deployed models.

You will bridge research and production, guide a small team, and deliver robust, low-latency systems while improving models over time. The role emphasizes end‑to‑end pipelines, optimization, and collaboration with engineers across back-end and front-end teams.

Qualifications

• Strong track record leading ML teams and delivering production models.
• Deep learning expertise with transformer architectures.
• Hands-on experience with PyTorch or JAX and distributed training tools.
• Ability to design robust data pipelines and scalable inference systems.

Responsibilities

• Own end-to-end ML pipelines from data to deployment.
• Architect scalable, low-latency inference systems.
• Lead model fine-tuning, evaluation, and deployment updates.
• Collaborate with engineers to integrate ML into products.

Skills

Team leadership

Deep learning

Transformer architectures

PyTorch

JAX

Distributed training

Code quality

GPU optimization

Ambiguity handling

Production ML

Tools

DeepSpeed

FSDP

Megatron

ZeRO

Ray

Job description

Digital Waffle is building a scalable ML infrastructure and seeks a hands-on leader to own the full ML lifecycle, from data to deployed models.

You will bridge research and production, guide a small team, and deliver robust, low-latency systems while improving models over time. The role emphasizes end‑to‑end pipelines, optimization, and collaboration with engineers across back-end and front-end teams.