Wayve

Applied Scientist Machine Learning Engineer Gaia (AI)

In short

This role involves developing efficient generative world models and planners for closed-loop driving simulation at an autonomous driving company. The work covers optimizing inference performance, diffusion and latent-video modeling, sim-to-real evaluation, and mentoring junior researchers. It requires 4+ years of ML research/engineering experience in generative video or world models, strong Python/PyTorch skills, and ideally a publication record or open-source contributions. Hybrid work schedule with core hours flexibility.

About the Role
You will develop efficient generative world models and planners for closed-loop driving simulation. You will optimize inference performance, define fidelity and coherence metrics, run experiments, integrate models into training and evaluation, measure sim-to-real performance, and mentor junior researchers.
Requirements
•4+ years of machine learning research or engineering experience focused on generative video or world models
•Knowledge of diffusion and latent-video models
•Experience improving sampling efficiency or model throughput
•Experience with high-dimensional temporal or spatiotemporal data
•Python and PyTorch engineering skills
•Experience building research-grade production tools
•Publication record or contributions to open-source machine-learning tooling
Responsibilities
•Develop efficient generative world models with real-time rollouts and controllable scene editing
•Architect interactive world models for reinforcement learning, planning, and safety evaluation
•Optimize latent compression, context pruning, and inference latency
•Define metrics for long-horizon coherence, physics fidelity, and planner integration
•Run ablation and scaling studies
•Integrate models into closed-loop training and evaluation
•Measure sim-to-real gaps against on-road driving-model results
•Mentor junior researchers and shape technical roadmaps
Benefits
•Hybrid working
•Core working hours with schedule flexibility

Stop applying.
Get referred.

$10 in free credits

In just 60 seconds, fully on autopilot — Eraldt's AI agents read your CV, map the roles you fit, find live vacancies and the people who can refer you, write personal emails, and get you referred.

This isn't an application: your CV doesn't go to the company, and we don't promise a referral for this exact role.

Discover who else lands interviews on autopilot