LICENSE
README.md
pyproject.toml
setup.py
benchmarks/__init__.py
benchmarks/atari_100k.py
tests/test_agents.py
tests/test_controller.py
tests/test_decoder.py
tests/test_determinism.py
tests/test_diamond.py
tests/test_dreamer.py
tests/test_encoder.py
tests/test_env_adapters.py
tests/test_envs.py
tests/test_iris_agent.py
tests/test_memory.py
tests/test_modular_rssm.py
tests/test_observations.py
tests/test_planet.py
tests/test_rewards.py
tests/test_sim.py
tests/test_trainers.py
tests/test_transforms.py
tests/test_ui_selenium.py
tests/test_utils.py
torchwm/__init__.py
torchwm.egg-info/PKG-INFO
torchwm.egg-info/SOURCES.txt
torchwm.egg-info/dependency_links.txt
torchwm.egg-info/not-zip-safe
torchwm.egg-info/requires.txt
torchwm.egg-info/top_level.txt
torchwm/sim/__init__.py
torchwm/sim/api.py
torchwm/sim/cli.py
torchwm/sim/gym_wrapper.py
torchwm/sim/perf.py
torchwm/sim/state.py
torchwm/sim/vector_env.py
torchwm/sim/worker.py
world_models/__init__.py
world_models/configs/__init__.py
world_models/configs/diamond_config.py
world_models/configs/dit_config.py
world_models/configs/dreamer_config.py
world_models/configs/iris_config.py
world_models/configs/jepa_config.py
world_models/controller/__init__.py
world_models/controller/iris_policy.py
world_models/controller/rollout_generator.py
world_models/controller/rssm_policy.py
world_models/datasets/__init__.py
world_models/datasets/cifar10.py
world_models/datasets/diamond_dataset.py
world_models/datasets/imagenet1k.py
world_models/datasets/nuplan.py
world_models/envs/__init__.py
world_models/envs/ale_atari_env.py
world_models/envs/ale_atari_vector_env.py
world_models/envs/diamond_atari.py
world_models/envs/dmc.py
world_models/envs/gym_env.py
world_models/envs/mujoco_env.py
world_models/envs/unity_env.py
world_models/envs/wrappers.py
world_models/memory/__init__.py
world_models/memory/dreamer_memory.py
world_models/memory/iris_memory.py
world_models/memory/planet_memory.py
world_models/models/__init__.py
world_models/models/dreamer.py
world_models/models/dreamer_rssm.py
world_models/models/iris_agent.py
world_models/models/iris_transformer.py
world_models/models/jepa_agent.py
world_models/models/modular_rssm.py
world_models/models/planet.py
world_models/models/rssm.py
world_models/models/vit.py
world_models/reward/__init__.py
world_models/reward/dreamer_v1_reward.py
world_models/reward/dreamer_v1_value.py
world_models/training/__init__.py
world_models/training/run_atari_100k.py
world_models/training/train_diamond.py
world_models/training/train_iris.py
world_models/training/train_jepa.py
world_models/training/train_planet.py
world_models/training/train_rssm.py
world_models/ui/__init__.py
world_models/ui/server.py
world_models/utils/__init__.py
world_models/utils/dreamer_utils.py
world_models/utils/jepa_utils.py
world_models/utils/utils.py
world_models/vision/__init__.py
world_models/vision/dreamer_decoder.py
world_models/vision/dreamer_encoder.py
world_models/vision/iris_decoder.py
world_models/vision/iris_encoder.py
world_models/vision/planet_decoder.py
world_models/vision/planet_encoder.py
world_models/vision/vq_layer.py