LeWAM: A JEPA World Action Model with Diffusion-Steering-Based MPC
LeWAM introduces a unified world action model that combines forward, backward, inverse dynamics, and policy prediction using a bidirectional transformer trained end-to-end. It achieves strong representational alignment by capturing dynamic states while ignoring visual distractors. This allows for superior planning via diffusion steering within the learned world model.