Senior AI Infra Engineer — Robotics Foundation Models

Axiōma Search; Greater London; Híbrido; GBP 110.000 - 170.000; Jornada completa

Descripción de la vacante

Axiōma Search is building a foundation model for manipulation tasks and looking for an hands-on IC to own distributed training end-to-end from London. This role involves scaling from tens to thousands of GPUs, improving efficiency and reliability, and shaping long-term infrastructure decisions.

You will work on-site in London five days a week, contributing code and leading the area independently as the team grows. Shortlisted candidates will be contacted quickly.

Formación

• Hands-on experience training large transformer models across multi-node GPU clusters.
• Deep distributed-training experience with PyTorch, NCCL, DeepSpeed, Megatron-LM/FSDP.
• Cloud-based GPU workloads, ideally AWS.

Responsabilidades

• Own distributed training infrastructure and scale it across large GPU clusters.
• Push training from tens of GPUs to hundreds, and potentially thousands.
• Improve GPU utilisation, throughput and end-to-end training efficiency.
• Optimise networking, storage, checkpointing and data pipelines for large jobs.
• Profile and optimise across the full training stack.
• Make long-term infrastructure and tooling decisions.
• Potentially support cloud inference and deployment as the product develops.

Conocimientos

Distributed training

Multi-node GPU clusters

Independent contributor

Hands-on coding

Herramientas

PyTorch

NCCL

DeepSpeed

Megatron-LM

FSDP

AWS

Descripción del empleo

Axiōma Search is building a foundation model for manipulation tasks and looking for an hands-on IC to own distributed training end-to-end from London. This role involves scaling from tens to thousands of GPUs, improving efficiency and reliability, and shaping long-term infrastructure decisions.

You will work on-site in London five days a week, contributing code and leading the area independently as the team grows. Shortlisted candidates will be contacted quickly.