Principal ML Engineer - Gaia World Model (London, Hybrid)

wayve; Greater London; Hybrid; GBP 120,000 - 180,000; Full time

Job summary

Wayve in London, United Kingdom, is seeking a Principal ML Engineer/Applied Scientist to own Gaia, Wayve's world model. You will drive post-training work and improve long-horizon autoregressive generation, guiding toward stable, steerable behaviours while collaborating across research, simulations, and cloud teams.

This is a hybrid, full-time role in London. You will lead training and fine-tuning of frontier-scale models, mentor engineers, and shape the next Gaia release in a fast-paced

Qualifications

• Hands-on experience post-training/fine-tuning large-scale models.
• Experience with world models, autoregressive generation, and long-horizon generation.
• Experience with diffusion/flow models and understanding of 3D vision.
• Strong understanding of model architecture and ability to contribute to architectural/training decisions.
• Strong hands-on engineering skills with modern ML stacks (e.g., PyTorch), including debugging and reliability-minded development.

Responsibilities

• Lead and own Gaia's post-training and closed-loop pipeline, fine-tuning world model.
• Improve Gaia's autoregressive generation for longer, stable rollouts and deploy-ready inference.
• Contribute to model architecture and training strategy decisions affecting downstream application.
• Collaborate with research, applications, simulation engineering, and cloud teams to translate post-training improvements into impact.
• Provide technical leadership through mentorship, reviews, and setting high engineering standards.

Skills

Large-scale models

World models

Autoregressive generation

Diffusion/flow models

3D vision

PyTorch

Architectural decisions

Mentorship

Debugging

Education

Advanced degree valued

Tools

PyTorch

Job description

Wayve in London, United Kingdom, is seeking a Principal ML Engineer/Applied Scientist to own Gaia, Wayve's world model. You will drive post-training work and improve long-horizon autoregressive generation, guiding toward stable, steerable behaviours while collaborating across research, simulations, and cloud teams.

This is a hybrid, full-time role in London. You will lead training and fine-tuning of frontier-scale models, mentor engineers, and shape the next Gaia release in a fast-paced