Research Engineer (Robotics - Model Training)
Axiōma Search is looking for a Research Engineer (Robotics - Model Training)
Job description
Research Engineer (Robotics - Model Training)
Robotics / AI Lab | London, Paris, or SF
Compensation: Up to £220k + equity package
About
This is an early-stage robotics + foundational AI company building universal robotics foundation models for general-purpose mobile robots. The ambition is to automate the large, under-served parts of the economy that traditional industrial robotics hasn’t touched.
They’re taking a vertical approach: simulation-first data generation, proprietary omni-models, tight hardware integration, and eventually custom silicon. The team is small, technical, and includes people behind some of the most significant advances in robotics and AI in recent years.
You’d be joining the team responsible for the core training stack.
What you'll do
- Build and optimise training infrastructure for large-scale vision-language and multimodal foundation models used across robotics tasks.
- Design systems for long-context video training, including sequence parallelism at scale
- Support autoregressive and diffusion-based models for actions and video, including real-time streaming inference
- Implement sampling during training (self-forcing) to reduce distribution drift
- Enable RL post-training for multimodal models
- Own data flow, memory movement, and GPU utilisation across complex training loops
What you'll need
- Extensive experience in ML infrastructure, distributed systems, and/or high-performance computing
- Direct experience training large vision-language or multimodal foundation models
- Strong background in large-scale distributed training and GPU performance tuning
- Experience from top AI labs, frontier model teams, or elite infrastructure groups
Shortlisted candidates will be contacted within 48 hours.
Extra information
- Status
- Open
- Education Level
- Secondary School
- Location
- London Area
- Type of Contract
- Full-time jobs
- Published at
- 09-02-2026
- Full UK/EU driving license preferred
- No
- Car Preferred
- No
- Must be eligible to work in the EU
- No
- Cover Letter Required
- No
- Languages
- English
Get similar vacancies sent to your mailbox
Fill in below which area you are searching in for a similar function and don't forget your e-mail address!