policy-learning

WorldDiT Tool

Four released checkpoints plus self-contained inference and evaluation code for a sub-billion-parameter diffusion transformer that emits continuous robot action chunks while predicting future camera-frame pixels as auxiliary training signal. The visual prediction head is dropped at deployment. Tested across four LIBERO simulation suites; the model card notes its cross-paper comparison mixes published protocols.