All modules for which code is available

• benchmarks.atari_100k
• world_models.configs.diamond_config
• world_models.configs.dit_config
• world_models.configs.dreamer_config
• world_models.configs.iris_config
• world_models.configs.jepa_config
• world_models.controller.iris_policy
• world_models.controller.rssm_policy
• world_models.datasets.cifar10
• world_models.datasets.imagenet1k
• world_models.envs.ale_atari_env
• world_models.envs.ale_atari_vector_env
• world_models.envs.dmc
• world_models.envs.gym_env
• world_models.envs.mujoco_env
• world_models.envs.unity_env
• world_models.envs.wrappers
• world_models.masks.multiblock
• world_models.masks.random
• world_models.memory.dreamer_memory
• world_models.memory.iris_memory
• world_models.memory.planet_memory
• world_models.models.diffusion.DDPM
• world_models.models.diffusion.DiT
• world_models.models.dreamer
• world_models.models.dreamer_rssm
• world_models.models.iris_agent
• world_models.models.iris_transformer
• world_models.models.vit
• world_models.training.train_iris
• world_models.transforms.transforms
• world_models.utils.dreamer_utils
• world_models.utils.jepa_utils
• world_models.utils.utils
• world_models.vision.dreamer_decoder
• world_models.vision.dreamer_encoder
• world_models.vision.iris_decoder
• world_models.vision.iris_encoder
• world_models.vision.vq_layer