LICENSE
README.md
pyproject.toml
setup.py
simulusion/__init__.py
simulusion/__init__.pyi
simulusion/_version.py
simulusion/api.py
simulusion/catalog.py
simulusion/cli.py
simulusion/export.py
simulusion/install_dmc.py
simulusion/py.typed
simulusion/registry.py
simulusion.egg-info/PKG-INFO
simulusion.egg-info/SOURCES.txt
simulusion.egg-info/dependency_links.txt
simulusion.egg-info/entry_points.txt
simulusion.egg-info/requires.txt
simulusion.egg-info/top_level.txt
simulusion/benchmarks/__init__.py
simulusion/benchmarks/adapters.py
simulusion/benchmarks/atari_100k.py
simulusion/benchmarks/cli.py
simulusion/benchmarks/metrics.py
simulusion/benchmarks/reporting.py
simulusion/benchmarks/runner.py
simulusion/benchmarks/tests/test_benchmarks_smoke.py
simulusion/blocks/__init__.py
simulusion/blocks/mhsa.py
simulusion/blocks/st_transformer.py
simulusion/configs/__init__.py
simulusion/configs/diamond_config.py
simulusion/configs/dit_config.py
simulusion/configs/dreamer_config.py
simulusion/configs/genie_config.py
simulusion/configs/iris_config.py
simulusion/configs/jepa_config.py
simulusion/configs/serialization.py
simulusion/configs/wm_config.py
simulusion/configs/experiments/__init__.py
simulusion/controller/__init__.py
simulusion/controller/iris_policy.py
simulusion/controller/rollout_generator.py
simulusion/controller/rssm_policy.py
simulusion/datasets/__init__.py
simulusion/datasets/cifar10.py
simulusion/datasets/diamond_dataset.py
simulusion/datasets/imagenet1k.py
simulusion/datasets/nuplan.py
simulusion/datasets/tinyworlds.py
simulusion/datasets/video_datasets.py
simulusion/datasets/wm_dataset.py
simulusion/envs/__init__.py
simulusion/envs/_actions.py
simulusion/envs/_contract.py
simulusion/envs/_observations.py
simulusion/envs/ale_atari_env.py
simulusion/envs/ale_atari_vector_env.py
simulusion/envs/brax_env.py
simulusion/envs/bsuite_env.py
simulusion/envs/diamond_atari.py
simulusion/envs/dmc.py
simulusion/envs/dmlab.py
simulusion/envs/gym_env.py
simulusion/envs/minecraft_env.py
simulusion/envs/mujoco_env.py
simulusion/envs/procgen_env.py
simulusion/envs/robotics_env.py
simulusion/envs/unity_env.py
simulusion/envs/vector_env.py
simulusion/envs/world_model_env.py
simulusion/envs/wrappers.py
simulusion/evals/__init__.py
simulusion/evals/diamond_utils.py
simulusion/evals/fid.py
simulusion/evals/fvd.py
simulusion/evals/lpips.py
simulusion/evals/psnr.py
simulusion/experiments/__init__.py
simulusion/experiments/config.py
simulusion/helpers/__init__.py
simulusion/helpers/jepa_helper.py
simulusion/inference/__init__.py
simulusion/inference/benchmark.py
simulusion/inference/bundle.py
simulusion/inference/eval_diamond.py
simulusion/inference/play_base.py
simulusion/inference/play_diamond.py
simulusion/inference/play_dreamer.py
simulusion/inference/precision.py
simulusion/inference/quantize.py
simulusion/inference/runtime.py
simulusion/inference/steppers.py
simulusion/layers/__init__.py
simulusion/layers/ada_ln_norm.py
simulusion/layers/rms_norm.py
simulusion/losses/__init__.py
simulusion/losses/convae_loss.py
simulusion/losses/gmm_loss.py
simulusion/masks/__init__.py
simulusion/masks/default.py
simulusion/masks/multiblock.py
simulusion/masks/random.py
simulusion/memory/__init__.py
simulusion/memory/dreamer_memory.py
simulusion/memory/iris_memory.py
simulusion/memory/planet_memory.py
simulusion/models/__init__.py
simulusion/models/controller.py
simulusion/models/dreamer.py
simulusion/models/dreamer_rssm.py
simulusion/models/dreamer_v1.py
simulusion/models/dreamer_v2.py
simulusion/models/dynamics_model.py
simulusion/models/genie.py
simulusion/models/iris_agent.py
simulusion/models/iris_transformer.py
simulusion/models/jepa_agent.py
simulusion/models/latent_action_model.py
simulusion/models/mdrnn.py
simulusion/models/model_io.py
simulusion/models/modular_rssm.py
simulusion/models/planet.py
simulusion/models/rssm.py
simulusion/models/vit.py
simulusion/models/diffusion/DDPM.py
simulusion/models/diffusion/DiT.py
simulusion/models/diffusion/__init__.py
simulusion/models/diffusion/actor_critic.py
simulusion/models/diffusion/diamond_diffusion.py
simulusion/models/diffusion/reward_termination.py
simulusion/observations/__init__.py
simulusion/observations/dreamer_v1_obs.py
simulusion/reward/__init__.py
simulusion/reward/dreamer_v1_reward.py
simulusion/reward/dreamer_v1_value.py
simulusion/testing/__init__.py
simulusion/testing/mocking_classes.py
simulusion/training/__init__.py
simulusion/training/eval_jepa.py
simulusion/training/rl_harness.py
simulusion/training/run_atari_100k.py
simulusion/training/train_controller.py
simulusion/training/train_convvae.py
simulusion/training/train_diamond.py
simulusion/training/train_dit.py
simulusion/training/train_dreamer.py
simulusion/training/train_genie.py
simulusion/training/train_iris.py
simulusion/training/train_jepa.py
simulusion/training/train_mdn_rnn.py
simulusion/training/train_planet.py
simulusion/training/train_rssm.py
simulusion/training/train_world_model.py
simulusion/training/train_world_model_carracing.py
simulusion/transforms/__init__.py
simulusion/transforms/image.py
simulusion/utils/__init__.py
simulusion/utils/data_utils.py
simulusion/utils/deprecation.py
simulusion/utils/device.py
simulusion/utils/dreamer_utils.py
simulusion/utils/gym_compat.py
simulusion/utils/jepa_utils.py
simulusion/utils/logging_utils.py
simulusion/utils/memory_utils.py
simulusion/utils/throughput.py
simulusion/utils/train_utils.py
simulusion/utils/utils.py
simulusion/vision/__init__.py
simulusion/vision/dreamer_decoder.py
simulusion/vision/dreamer_encoder.py
simulusion/vision/iris_decoder.py
simulusion/vision/iris_encoder.py
simulusion/vision/perceptual_loss.py
simulusion/vision/planet_decoder.py
simulusion/vision/planet_encoder.py
simulusion/vision/video_tokenizer.py
simulusion/vision/vq_layer.py
simulusion/vision/VAE/ConvVAE.py
simulusion/vision/VAE/__init__.py
tests/test_audit_regressions.py
tests/test_benchmark_infer.py
tests/test_config_serialization.py
tests/test_demo_scripts.py
tests/test_determinism_regressions.py
tests/test_diamond_inference.py
tests/test_diamond_web_export.py
tests/test_dreamer_integration.py
tests/test_dreamer_rssm.py
tests/test_experiment_config.py
tests/test_genie_integration.py
tests/test_inference_runtime.py
tests/test_iris_integration.py
tests/test_jepa_integration.py
tests/test_optional_backends_real.py
tests/test_optional_dependencies.py
tests/test_optional_imports.py
tests/test_public_api.py
tests/test_readme_algorithms.py
tests/test_release_workflow.py
tests/test_type_stub.py
tests/test_world_model_env.py