Principal Machine Learning Engineer GAIA

Job summary

Responsibilities

You'll be joining the Simulation team (Simulation Evaluation & Validation), working on Gaia — Wayve's world model. Gaia is trained on large-scale driving video to predict future frames from past context, functioning as a simulator that generates synthetic scenarios and operates in closed loop with…

• Own and lead parts of large-scale training for Gaia (language- or video-style training at scale), from experimentation through productionised training runs.
• Contribute to and influence model architecture decisions — not just applying existing models.
• Partner across sub-teams: Research (model development), Applications (adapting models to use cases), Core Simulation Engineering, and Cloud/Infrastructure.
• Translate product and evaluation needs into model improvements (e.g. better long-horizon prediction, scenario generation quality, failure-case coverage).
• Mentor and set technical direction, raising the engineering and research quality bar across the team.

Tech stack

• PyTorch