LeMario: Super Mario Bros trained on a JEPA Model
Summary
LeMario is a JEPA-based world model trained on Super Mario Bros. The article presents the architecture, experiments, results, and lessons learned, including the model's ability to predict short-horizon dynamics and the challenges of planning with latent representations. It discusses SIGReg, AdaLN-Zero, CEM planning, and the gap between predictive state and controllable state, along with future research directions.