Deep Learning Engineer - World Models

As a Research Engineer on the World Models team, you will build action-conditioned generative models that predict how the world evolves around our robots — future video, proprioception, contacts, and outcomes — from past observations and actions. World models serve four purposes in our stack: a pretrained, physics-aware prior for our VLA policies; an engine for rare data collection, cross-platform transfer, and sim-to-real transfer; a testbed for policy evaluation and testing before hardware; and a future-prediction rollout engine that surfaces what our policies intend to do, for safety and planning.

This is a hands-on individual contributor role: you will design architectures, run large training jobs, and validate your models against real fleet data from industrial deployments.

• Design and train multimodal world models — video, state, action, and language — using diffusion-based and transformer architectures.