Staff ML Engineer, Gaia
Core
Own and drive work on training and improving frontier-scale video world models (Gaia) that predict future frames to generate synthetic driving scenarios for automated driving systems.
Role type
Staff ML Engineer (Video World Models)
Builds
Gaia video world model, synthetic driving scenarios, and downstream evaluation/training pipelines for driving models
Domain
Autonomous driving, Embodied AI, Video foundation models
Deliverable
production ML models
Required skills
Large-scale model training, model architecture design, PyTorch, distributed training, debugging, performance optimization
Preferred skills
World models, video generation, long-horizon prediction, data pipeline improvement, technical leadership
Technologies
PyTorch
Responsibilities
Lead large-scale training runs for video foundation models, contribute to model architecture and training strategy, improve world-model capabilities for synthetic scenario generation, partner with research and infrastructure teams, provide technical leadership and mentorship
Seniority
Staff, hands-on IC with tech-lead scope