Up to £180k base + equity package
About
A VC-backed robotics startup in London is building a transformer-based foundation model for robotic manipulation. The team is fewer than ten people, technically ambitious and highly hands-on, with most training data generated through simulation.
Their next challenge is scaling training from the current cloud setup to hundreds of GPUs. That introduces a different class of engineering problems: distributed communication across nodes, reliable checkpointing, high-throughput storage and data loading, GPU utilisation, observability and compute cost.
They are looking for a Research Engineer to take ownership of this scaling work, improving both training speed and compute efficiency. You do not need to have solved every problem at this scale already. Strong ML infrastructure foundations, an understanding of how distributed systems begin to fail under load, and the appetite to learn quickly in an early-stage environment matter more than extensive experience.
What you'll do
- Build and operate the infrastructure behind multi-node GPU training, including job orchestration, scheduling, environments and recovery
- Make training reliable, so a failed node costs minutes rather than days
- Profile the stack end to end and remove the constraints limiting throughput
- Solve the less glamorous but critical problems across checkpointing, data loading, storage, networking and image builds
- Build the observability the team currently lacks, making utilisation, performance and cost visible
- Work directly with researchers and take infrastructure problems off their plate
- Grow into decisions about how a large compute budget is allocated and committed
What you'll need
- Strong Python and PyTorch
- Experience building ML infrastructure, with some exposure to multi-node GPU training
- A working understanding of distributed training, including NCCL, FSDP or DeepSpeed, and where these systems tend to break
- Experience working with cloud GPUs, ideally on AWS
- Willingness to debug below the orchestration layer across communications, storage, I/O and hardware
- A genuine appetite for the pace, ownership and ambiguity of an early-stage startup
Bonus
- Experience supporting researchers or research engineers directly
- Kernel or performance work using CUDA or Triton
- Deeper expertise in storage or networking
- Open-source contributions to training or inference infrastructure
Shortlisted candidates will be contacted within 48 hours.
#J-18808-Ljbffr
Research Engineer employer: Axiōma Search
Axiōma Search is an exceptional employer for those passionate about machine learning and time-series forecasting. With a collaborative work culture that prioritises innovation and employee growth, team members are encouraged to explore new ideas and technologies while benefiting from a global network of experts. Located in Europe, the company offers unique opportunities for professional development and the chance to make a significant impact in the field of AI.
