Lightweight implementation of Robotic World Model (RWM) and Uncertainty-Aware Robotic World Model (RWM-U) - View it on GitHub