LeWAM: End-to-End World and Action Modeling with JEPAs

LeWAM is an end-to-end JEPA for control that jointly trains next latent state prediction and action flow matching with an isotropic Gaussian constraint on the latent state, improving success rates on contact-rich and long-horizon tasks while planning faster than prior JEPAs.