Principal Machine Learning Engineer GAIA

About the role

You will join the Simulation team to work on Gaia, Wayve's world model trained on large-scale driving video to predict future frames and generate synthetic scenarios. This role involves leading large-scale training, influencing model architecture, and partnering with research and infrastructure teams to turn model improvements into measurable impact on the driving stack.

Responsibilities

• Own and lead parts of large-scale training for Gaia from experimentation through productionised training runs.
• Contribute to and influence model architecture decisions beyond applying existing models.
• Translate product and evaluation needs into model improvements such as long-horizon prediction and failure-case coverage.
• Mentor and set technical direction to raise engineering and research quality across the team.
• Lead and execute Gaia's post-training and closed-loop pipeline including fine-tuning and alignment.
• Push autoregressive generation towards longer, more stable rollouts and ensure model deployment readiness.

Required skills

• Post-training
• Fine-tuning
• World models
• Autoregressive generation
• Long-horizon generation
• Diffusion models
• Flow models
• 3D vision
• Model architecture
• PyTorch
• Simulation
• Evaluation
• Validation

Nice to have

• Inference optimisation
• Distributed training
• Technical leadership

Qualifications

• 5+ years of relevant industry experience

Benefits

• Health insurance
• Dental insurance
• Enhanced maternity leave
• Enhanced paternity leave
• Retirement or pension
• Access to therapists
• Wellbeing partnerships
• Learning and development budget
• Equity

About the Company

Wayve is building the leading AI platform for autonomous driving, pioneering an end-to-end AI approach that enables vehicles to learn directly from real-world experience. The company's mapless and hardware-agnostic AI platform integrates with global OEM partners to enable continuous software evolution and advanced levels of automation.