LeWAM: end-to-end JEPA lifts contact-rich task success from 28.6% to 89.7%, plans 32x faster

LeWAM · hf · 2026-10-14

Researchers from UC San Diego, ETH Zurich, UBC, and Brown introduce LeWAM, an end-to-end JEPA for control that jointly trains next-latent-state prediction with action flow matching under an isotropic Gaussian latent constraint, letting one model both execute policies directly and plan by scoring sampled actions. In visual planning it lifts contact-rich task success from 28.6% to 89.7% and long-horizon from 10.9% to 46.2%, while planning 32.3x faster; it also matches established methods on behavior cloning in simulation and on physical robots. Code and website are available.

Original post →

More from Embodied

Embodied channel →