Principal Machine Learning Engineer GAIA
Core
Own and drive training and improvement of Gaia, Wayve's world model, a frontier-scale model trained on driving video that predicts future frames to simulate scenarios and operate in closed loop with the driving model.
Role type
Principal Machine Learning Engineer / Applied Scientist (World Models)
Builds
Gaia world model (autoregressive video generation, long-horizon rollouts, closed-loop simulation)
Domain
Autonomous driving, embodied intelligence, world models
Deliverable
production ML models
Required skills
post-training/fine-tuning large-scale foundation models, world models, autoregressive generation, long-horizon generation, diffusion/flow models, 3D vision, model architecture design, PyTorch engineering, debugging, performance optimization
Preferred skills
inference optimization under latency constraints, distributed training, data pipeline improvement, technical leadership
Technologies
PyTorch
Responsibilities
Lead post-training and closed-loop pipeline execution, push autoregressive generation for stable long rollouts, contribute to model architecture and training strategy decisions, partner with research and infrastructure teams, provide technical leadership and mentorship
Seniority
Principal, hands-on IC with tech-lead ownership
