galilai-group / stable-worldmodel Python 2.3K 295 A platform for reproducible world model research and evaluation deep-learning pytorch jepa world-model