Synthetic Data ML Engineer for World Models

AI Startups UK; Greater London; Hybrid; GBP 90,000 - 140,000; Full time

Benefits offered by this job

Hybrid working

Office in London

Job summary

Wayve, based in London, seeks an ML/Research Engineer to build and scale GAIA-class world models and bridge them into high-throughput synthetic data generation. You will work between ML research and engineering, delivering training-ready artefacts and maintaining data lineage.

You will post-train world models, generate multimodal synthetic experiences, and integrate them into the training stack used for real driving.

Qualifications

• 4+ years in applied ML / research engineering, with a track record of training and shipping neural nets.
• Strong Python and PyTorch; comfortable with GPU training, debugging, and reading model code.
• Hands-on with video, generative, or world models (diffusion / flow-matching / autoregressive video).
• Working knowledge of cameras and 3D geometry (multi-camera rigs, intrinsics/extrinsics, warps) and why they break generation.
• Evidence of taking generated or simulated data into a trained downstream model and measuring impact.
• Ability to operate generation/training at real scale (multi-GPU; workflow orchestration) and make it reliable.
• Collaborative, experimental working style; you will own a capability, not a ticket queue.

Responsibilities

• Post-train and iterate GAIA-class world models for synthetic-data capabilities (rig transfer, pose transfer, conditioning).
• Own the generation loop: config → large-scale GPU inference → training-ready artefacts with clear lineage.
• Land synthetic data in driving-model training (behavior cloning, reward models, RL) with quality filters and experiments.
• Diagnose and fix geometry, calibration, and controllability failures affecting training data quality.
• Improve throughput and yield: inference optimisations, valid-generation rate, and self-serve workflows.
• Expand coverage to new vehicle platforms and safety-critical scenarios (OEM bring-up; Emergency Lane Keeping / AEB).
• Partner with researchers, infra, and driving-model owners to keep generation, evaluation, training as a single system.

Skills

Applied ML / research engineering

Python

PyTorch

GPU training

World models / generative video

Cameras & 3D geometry

Synthetic data generation

Tools

Flyte

Ray

Spark

Azure

AWS

GCP

Job description

Wayve, based in London, seeks an ML/Research Engineer to build and scale GAIA-class world models and bridge them into high-throughput synthetic data generation. You will work between ML research and engineering, delivering training-ready artefacts and maintaining data lineage.

You will post-train world models, generate multimodal synthetic experiences, and integrate them into the training stack used for real driving.