Index A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | Q | R | S | T | U | V | W A act() (simulusion.agents.planning.PlanningAgent method) (simulusion.controller.iris_policy.IRISPolicy method) (simulusion.inference.steppers.DreamerStepper method) (simulusion.inference.steppers.IRISStepper method) (simulusion.inference.steppers.WorldModelStepper method) (simulusion.models.iris_agent.IRISAgent method) act_and_collect_data() (simulusion.models.dreamer.Dreamer method) act_with_world_model() (simulusion.models.dreamer.Dreamer method) action (simulusion.data.rlds.RLDSKeys attribute) (simulusion.world_model.protocol.WorldModelSpec attribute) (simulusion.world_model.WorldModelSpec attribute) action_decoder_dim (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) action_dim (simulusion.data.folder.DatasetMeta attribute) action_embedding_dim (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) action_encoder_depth (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) action_encoder_dim (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) action_high (simulusion.data.folder.DatasetMeta attribute) action_kind (simulusion.data.folder.DatasetMeta attribute) action_low (simulusion.data.folder.DatasetMeta attribute) action_noise (simulusion.configs.dreamer_config.DreamerConfig attribute) action_num_heads (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) action_pooling (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.configs.genie_config.LatentActionModelConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) action_repeat (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) action_size (simulusion.configs.wm_config.WMControllerConfig attribute) (simulusion.configs.wm_config.WMMDNRNNConfig attribute) (simulusion.utils.utils.TorchImageEnvWrapper property) action_space (simulusion.envs.dmc.DeepMindControlEnv property) (simulusion.envs.dmlab.DMLabEnv property) (simulusion.envs.gym_env.GymImageEnv property) (simulusion.envs.mujoco_env.MuJoCoImageEnv property) (simulusion.envs.procgen_env.ProcgenImageEnv property) (simulusion.envs.unity_env.UnityMLAgentsEnv property) (simulusion.envs.wrappers.FrameStack property) (simulusion.envs.wrappers.NormalizeActions property) (simulusion.envs.wrappers.ObsDict property) (simulusion.envs.wrappers.OneHotAction property) action_vocab_size (simulusion.configs.genie_config.DynamicsModelConfig attribute) (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) ActionDecoder (class in simulusion.vision.dreamer_decoder) ActionRepeat (class in simulusion.envs.wrappers) actions (simulusion.agents.planning.PlanningAgent attribute) (simulusion.data.episode.Episode attribute) ActionSpec (class in simulusion.world_model) (class in simulusion.world_model.protocol) actor_channels (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.diamond_config.ModelPreset attribute) actor_critic_batch_size (simulusion.configs.iris_config.IRISConfig attribute) actor_critic_steps_per_epoch (simulusion.configs.iris_config.IRISConfig attribute) actor_hidden_size (simulusion.configs.iris_config.IRISConfig attribute) actor_layers (simulusion.configs.iris_config.IRISConfig attribute) actor_learning_rate (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) actor_loss() (simulusion.models.dreamer.Dreamer method) actor_lstm_dim (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.diamond_config.ModelPreset attribute) actor_res_blocks (simulusion.configs.diamond_config.DiamondConfig attribute) ActorCritic (class in simulusion.training.rl_harness) ActorCriticNetwork (class in simulusion.models.diffusion.actor_critic) adam_beta1 (simulusion.configs.iris_config.IRISConfig attribute) adam_beta2 (simulusion.configs.iris_config.IRISConfig attribute) adam_epsilon (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.dreamer_config.DreamerConfig attribute) AdaptiveGroupNorm (class in simulusion.models.diffusion.diamond_diffusion) add() (simulusion.datasets.diamond_dataset.ReplayBuffer method) (simulusion.memory.dreamer_memory.ReplayBuffer method) (simulusion.memory.iris_memory.IRISOnPolicyBuffer method) (simulusion.memory.iris_memory.IRISReplayBuffer method) add_exploration() (simulusion.vision.dreamer_decoder.ActionDecoder method) advance() (simulusion.blocks.st_transformer.STKVCache method) (simulusion.models.iris_transformer.KVCache method) agent_types (simulusion.datasets.nuplan.NuPlanSample attribute) agents_future (simulusion.datasets.nuplan.NuPlanSample attribute) agents_mask (simulusion.datasets.nuplan.NuPlanSample attribute) agents_past (simulusion.datasets.nuplan.NuPlanSample attribute) algo (simulusion.configs.dreamer_config.DreamerConfig attribute) algo_name (simulusion.models.dreamer.DreamerAgent attribute) aliases (simulusion.api.EnvBackendSpec attribute) (simulusion.api.ModelSpec attribute) AllGather (class in simulusion.utils.jepa_utils) allow_overlap (simulusion.configs.jepa_config.JEPAConfig attribute) AllReduce (class in simulusion.utils.jepa_utils) AllReduceSum (class in simulusion.utils.jepa_utils) append() (simulusion.blocks.st_transformer.STKVCache method) (simulusion.memory.dreamer_memory.Episode method) (simulusion.memory.dreamer_memory.Memory method) (simulusion.memory.planet_memory.Episode method) (simulusion.memory.planet_memory.Memory method) (simulusion.models.iris_transformer.KVCache method) apply_gradient_checkpointing() (in module simulusion.utils.memory_utils) apply_masks() (in module simulusion.utils.utils) apply_model() (in module simulusion.utils.utils) artifacts (simulusion.inference.bundle.Bundle attribute) as_dict() (simulusion.inference.benchmark.DriftReport method) (simulusion.inference.benchmark.LatencyReport method) as_frames() (simulusion.world_model.protocol.WorldModelBase method) (simulusion.world_model.WorldModelBase method) as_tensor() (simulusion.world_model.ActionSpec method) (simulusion.world_model.protocol.ActionSpec method) as_uint8_frames() (in module simulusion.data.episode) as_world_model() (in module simulusion.world_model) aspect_ratio (simulusion.configs.jepa_config.JEPAConfig attribute) assert_finite() (in module simulusion.utils.logging_utils) assert_finite_values() (in module simulusion.utils.logging_utils) assign_type() (simulusion.utils.utils.TensorBoardMetrics method) atanh() (simulusion.vision.dreamer_decoder.TanhBijector method) atari_100k (simulusion.configs.iris_config.IRISConfig attribute) attach() (simulusion.training.task.TrainingTask method) Attention (class in simulusion.models.vit) AttentionBlock (class in simulusion.models.diffusion.diamond_diffusion) attn_drop_rate (simulusion.configs.genie_config.DynamicsModelConfig attribute) (simulusion.configs.genie_config.STTransformerConfig attribute) AttrDict (class in simulusion.utils.utils) autocast() (simulusion.training.precision.MixedPrecision method) autoencoder_batch_size (simulusion.configs.iris_config.IRISConfig attribute) autoregressive_sample() (simulusion.models.dynamics_model.DynamicsModel method) AverageMeter (class in simulusion.utils.jepa_utils) B BackboneBase (class in simulusion.models.modular_rssm) backward() (simulusion.utils.jepa_utils.AllGather static method) (simulusion.utils.jepa_utils.AllReduce static method) (simulusion.utils.jepa_utils.AllReduceSum static method) BaseAdapter (class in simulusion.benchmarks.adapters) BATCH (simulusion.configs.dit_config.DiTConfig attribute) batch_size (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.configs.jepa_config.JEPAConfig attribute) (simulusion.configs.wm_config.WMMDNRNNConfig attribute) (simulusion.datasets.tinyworlds.TinyWorldsConfig attribute) (simulusion.datasets.video_datasets.DatasetConfig attribute) (simulusion.inference.benchmark.LatencyReport attribute) (simulusion.training.tasks.common.SequenceDataConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) batch_size() (simulusion.world_model.protocol.WorldModelBase static method) (simulusion.world_model.WorldModelBase static method) benchmark_step() (in module simulusion.inference.benchmark) BenchmarkRunner (class in simulusion.benchmarks.runner) best_checkpoint (simulusion.training.trainer.TrainResult attribute) best_metric (simulusion.training.trainer.TrainerConfig attribute) best_mode (simulusion.training.trainer.TrainerConfig attribute) best_value (simulusion.training.trainer.TrainResult attribute) BETA_END (simulusion.configs.dit_config.DiTConfig attribute) BETA_START (simulusion.configs.dit_config.DiTConfig attribute) Block (class in simulusion.models.vit) bootstrap_ci() (in module simulusion.benchmarks.metrics) bootstrap_iqm_ci() (in module simulusion.benchmarks.metrics) bottle() (in module simulusion.utils.utils) brax_auto_reset (simulusion.configs.dreamer_config.DreamerConfig attribute) brax_backend (simulusion.configs.dreamer_config.DreamerConfig attribute) brax_jit (simulusion.configs.dreamer_config.DreamerConfig attribute) brax_suppress_warp_warnings (simulusion.configs.dreamer_config.DreamerConfig attribute) buffer_capacity (simulusion.memory.iris_memory.IRISReplayBuffer property) buffer_size (simulusion.configs.dreamer_config.DreamerConfig attribute) build_diamond_networks() (in module simulusion.world_model.diamond) build_genie() (in module simulusion.training.train_genie) build_loss_fn() (in module simulusion.training.train_jepa) build_parser() (in module simulusion.training.train_planet) Bundle (class in simulusion.inference.bundle) burn_in() (simulusion.models.iris_agent.IRISAgent method) burn_in_length (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) C cache_dir (simulusion.datasets.tinyworlds.TinyWorldsConfig attribute) camera (simulusion.training.tasks.common.SequenceDataConfig attribute) cameras (simulusion.data.episode.Episode attribute) (simulusion.data.folder.DatasetMeta attribute) (simulusion.data.rlds.RLDSKeys attribute) candidates (simulusion.agents.planning.CEMPlanner attribute) canonical_dit_key() (in module simulusion.configs.dit_config) CausalSelfAttention (class in simulusion.models.iris_transformer) CEMPlanner (class in simulusion.agents.planning) CHANNELS (simulusion.configs.dit_config.DiTConfig attribute) checkpoint (simulusion.training.trainer.TrainResult attribute) checkpoint_dir (simulusion.configs.diamond_config.DiamondConfig attribute) CHECKPOINT_EVERY (simulusion.configs.dit_config.DiTConfig attribute) checkpoint_every (simulusion.training.trainer.TrainerConfig attribute) CHECKPOINT_FORMAT (simulusion.models.genie.Genie attribute) (simulusion.models.iris_agent.IRISAgent attribute) checkpoint_interval (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) checkpoint_path (simulusion.configs.dreamer_config.DreamerConfig attribute) CLASS_DROPOUT_PROB (simulusion.configs.dit_config.DiTConfig attribute) classes (simulusion.datasets.imagenet1k.ImageNetSubset property) clear() (simulusion.memory.iris_memory.IRISOnPolicyBuffer method) close() (simulusion.data.folder.EpisodeWriter method) (simulusion.envs.diamond_atari.DiamondAtariWrapper method) (simulusion.envs.dmlab.DMLabEnv method) (simulusion.envs.gym_env.GymImageEnv method) (simulusion.envs.mujoco_env.MuJoCoImageEnv method) (simulusion.envs.procgen_env.ProcgenImageEnv method) (simulusion.envs.unity_env.UnityMLAgentsEnv method) (simulusion.envs.vector_env.VectorizedEnv method) (simulusion.training.task.TrainingTask method) (simulusion.utils.logging_utils.MetricsLogger method) (simulusion.utils.utils.StreamingVideoWriter method) (simulusion.utils.utils.TorchImageEnvWrapper method) cnn_activation_function (simulusion.configs.dreamer_config.DreamerConfig attribute) CNNDecoder (class in simulusion.vision.planet_decoder) CNNEncoder (class in simulusion.vision.planet_encoder) CNNFeatureExtractor (class in simulusion.controller.iris_policy) collate_sequences() (in module simulusion.data.sequence) collect_epsilon (simulusion.configs.iris_config.IRISConfig attribute) collect_experience() (simulusion.training.train_iris.IRISTrainer method) collect_random_episodes() (simulusion.models.dreamer.Dreamer method) collect_steps (simulusion.configs.dreamer_config.DreamerConfig attribute) collect_system_stats() (in module simulusion.utils.logging_utils) collect_temperature (simulusion.configs.iris_config.IRISConfig attribute) collect_trajectories() (simulusion.training.rl_harness.PPOTrainer method) collection_epochs (simulusion.configs.iris_config.IRISConfig attribute) color_jitter_strength (simulusion.configs.jepa_config.JEPAConfig attribute) combine_videos() (in module simulusion.utils.utils) commitment_weight (simulusion.configs.genie_config.LatentActionModelConfig attribute) (simulusion.configs.genie_config.VideoTokenizerConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) compute_aggregate_metrics() (in module simulusion.benchmarks.metrics) compute_gae() (simulusion.training.rl_harness.PPOTrainer method) compute_lambda_return() (in module simulusion.models.iris_agent) compute_lambda_returns() (simulusion.models.diffusion.actor_critic.RLLoss method) compute_return() (in module simulusion.utils.dreamer_utils) config (simulusion.inference.bundle.Bundle attribute) config_cls (simulusion.training.task.TrainingTask attribute) config_path (simulusion.api.ModelSpec attribute) console (simulusion.training.trainer.TrainerConfig attribute) context_frames (simulusion.world_model.protocol.WorldModelSpec attribute) (simulusion.world_model.WorldModelSpec attribute) continue_prob (simulusion.inference.steppers.ImagineOutput attribute) Controller (class in simulusion.models.controller) conv_vae_loss_fn() (in module simulusion.losses.convae_loss) ConvDecoder (class in simulusion.models.modular_rssm) (class in simulusion.vision.dreamer_decoder) ConvEmbed (class in simulusion.models.vit) ConvEncoder (class in simulusion.models.modular_rssm) (class in simulusion.vision.dreamer_encoder) ConvVAE (class in simulusion.vision.VAE.ConvVAE) ConvVAEDecoder (class in simulusion.vision.VAE.ConvVAE) ConvVAEEncoder (class in simulusion.vision.VAE.ConvVAE) copy_data (simulusion.configs.jepa_config.JEPAConfig attribute) copy_imgnt_locally() (in module simulusion.datasets.imagenet1k) core_cls (simulusion.models.dreamer.DreamerAgent attribute) CosineWDSchedule (class in simulusion.utils.jepa_utils) create_config() (in module simulusion.api) create_dataloader() (simulusion.datasets.tinyworlds.TinyWorldsDataLoader static method) create_dit() (in module simulusion.models.diffusion.DiT) create_dynamics_model() (in module simulusion.models.dynamics_model) create_efficient_dataloader() (in module simulusion.utils.data_utils) create_genie() (in module simulusion.models.genie) create_genie_large() (in module simulusion.models.genie) create_genie_small() (in module simulusion.models.genie) create_genie_trainer() (in module simulusion.training.train_genie) create_latent_action_model() (in module simulusion.models.latent_action_model) create_model() (in module simulusion.api) create_modular_rssm() (in module simulusion.models.modular_rssm) create_rl_harness_example() (in module simulusion.training.rl_harness) create_st_transformer() (in module simulusion.blocks.st_transformer) create_tinyworlds_dataloader() (in module simulusion.datasets.tinyworlds) create_video_dataloader() (in module simulusion.datasets.video_datasets) create_video_dataset_from_config() (in module simulusion.datasets.video_datasets) create_video_tokenizer() (in module simulusion.vision.video_tokenizer) crop_scale (simulusion.configs.jepa_config.JEPAConfig attribute) CROP_SIZE (simulusion.configs.dit_config.DiTConfig attribute) crop_size (simulusion.configs.jepa_config.JEPAConfig attribute) CSVLogger (class in simulusion.utils.jepa_utils) D data_dir (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.wm_config.WMMDNRNNConfig attribute) (simulusion.configs.wm_config.WMVAEConfig attribute) data_loader_num_workers (simulusion.configs.diamond_config.DiamondConfig attribute) data_source (simulusion.datasets.video_datasets.VideoDatasetBase attribute) (simulusion.datasets.video_datasets.VideoDatasetConfig attribute) DATASET (simulusion.configs.dit_config.DiTConfig attribute) dataset (simulusion.configs.jepa_config.JEPAConfig attribute) DATASET_CONFIGS (simulusion.datasets.tinyworlds.TinyWorldsDataset attribute) dataset_name (simulusion.datasets.tinyworlds.TinyWorldsConfig attribute) DATASET_NAMES (simulusion.datasets.tinyworlds.TinyWorldsDataLoader attribute) dataset_type (simulusion.datasets.video_datasets.VideoDatasetConfig attribute) DatasetConfig (class in simulusion.datasets.video_datasets) DatasetMeta (class in simulusion.data.folder) DDPM (class in simulusion.models.diffusion.DDPM) decode() (simulusion.inference.steppers.DreamerStepper method) (simulusion.models.latent_action_model.LatentActionModel method) (simulusion.utils.dreamer_utils.TwoHotEncoder method) (simulusion.vision.iris_decoder.DiscreteAutoencoder method) (simulusion.vision.video_tokenizer.VideoTokenizer method) (simulusion.world_model.protocol.WorldModel method) (simulusion.world_model.protocol.WorldModelBase method) (simulusion.world_model.WorldModel method) (simulusion.world_model.WorldModelBase method) decode_from_embeddings() (simulusion.vision.iris_decoder.IRISDecoder method) decode_from_indices() (simulusion.vision.iris_decoder.IRISDecoder method) (simulusion.vision.iris_encoder.IRISEncoder method) decode_indices() (simulusion.vision.video_tokenizer.VideoTokenizer method) (simulusion.vision.vq_layer.VectorQuantizer method) (simulusion.vision.vq_layer.VectorQuantizerEMA method) decode_observation() (simulusion.models.modular_rssm.ModularRSSM method) decode_reward() (simulusion.models.modular_rssm.ModularRSSM method) decode_tokens() (simulusion.models.iris_transformer.IRISWorldModel method) decoder_depth (simulusion.configs.genie_config.VideoTokenizerConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) decoder_dim (simulusion.configs.genie_config.VideoTokenizerConfig attribute) DecoderBase (class in simulusion.models.modular_rssm) DeepMindControlEnv (class in simulusion.envs.dmc) default_collect_temperature() (in module simulusion.training.train_iris) default_max_steps (simulusion.training.task.TrainingTask attribute) DefaultCollator (class in simulusion.masks) (class in simulusion.masks.default) denoise() (simulusion.models.diffusion.diamond_diffusion.EDMPreconditioner method) dense_activation_function (simulusion.configs.dreamer_config.DreamerConfig attribute) DenseDecoder (class in simulusion.vision.dreamer_decoder) DEPTH (simulusion.configs.dit_config.DiTConfig attribute) depth (simulusion.configs.genie_config.DynamicsModelConfig attribute) (simulusion.configs.genie_config.STTransformerConfig attribute) dequantized_weight() (simulusion.inference.quantize.Int8WeightOnlyLinear method) description (simulusion.api.EnvBackendSpec attribute) (simulusion.api.ModelSpec attribute) detach_state() (simulusion.models.dreamer_rssm.RSSM method) (simulusion.models.modular_rssm.ModularRSSM method) detect_anomaly (simulusion.configs.dreamer_config.DreamerConfig attribute) deter_size (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.models.modular_rssm.BackboneBase attribute) (simulusion.models.modular_rssm.ModularRSSM property) deterministic (simulusion.training.trainer.TrainerConfig attribute) deterministic_state_fwd() (simulusion.models.rssm.RecurrentStateSpaceModel method) device (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.wm_config.WMMDNRNNConfig attribute) (simulusion.configs.wm_config.WMVAEConfig attribute) (simulusion.inference.runtime.InferenceModel property) (simulusion.inference.steppers.WorldModelStepper attribute) (simulusion.training.trainer.Trainer attribute) (simulusion.training.trainer.TrainerConfig attribute) (simulusion.world_model.protocol.WorldModel attribute) (simulusion.world_model.protocol.WorldModelBase attribute) (simulusion.world_model.WorldModel attribute) (simulusion.world_model.WorldModelBase attribute) DeviceLoader (class in simulusion.training.trainer) DiamondAdapter (class in simulusion.benchmarks.adapters) DiamondAgent (class in simulusion.training.train_diamond) DiamondAtariWrapper (class in simulusion.envs.diamond_atari) DiamondConfig (class in simulusion.configs.diamond_config) DiamondWorldModel (class in simulusion.world_model.diamond) diffusion_channels (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.diamond_config.ModelPreset attribute) diffusion_cond_dim (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.diamond_config.ModelPreset attribute) diffusion_res_blocks (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.diamond_config.ModelPreset attribute) DiffusionUNet (class in simulusion.models.diffusion.diamond_diffusion) dim (simulusion.configs.genie_config.DynamicsModelConfig attribute) (simulusion.configs.genie_config.STTransformerConfig attribute) disc_loss_coeff (simulusion.configs.dreamer_config.DreamerConfig attribute) discount (simulusion.agents.planning.CEMPlanner attribute) (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) discount_factor (simulusion.configs.diamond_config.DiamondConfig attribute) DiscreteAutoencoder (class in simulusion.vision.iris_decoder) display (simulusion.configs.wm_config.WMControllerConfig attribute) DiT (class in simulusion.models.diffusion.DiT) dit_ema_update() (in module simulusion.models.diffusion.DiT) dit_l_simple() (in module simulusion.models.diffusion.DiT) dit_loaders() (in module simulusion.models.diffusion.DiT) dit_preset_config() (in module simulusion.configs.dit_config) DiTConfig (class in simulusion.configs.dit_config) dmlab_action_repeat (simulusion.configs.dreamer_config.DreamerConfig attribute) dmlab_action_set (simulusion.configs.dreamer_config.DreamerConfig attribute) dmlab_config (simulusion.configs.dreamer_config.DreamerConfig attribute) dmlab_observations (simulusion.configs.dreamer_config.DreamerConfig attribute) dmlab_renderer (simulusion.configs.dreamer_config.DreamerConfig attribute) DMLabEnv (class in simulusion.envs.dmlab) DownBlock (class in simulusion.models.diffusion.diamond_diffusion) download (simulusion.configs.jepa_config.JEPAConfig attribute) download_all_datasets() (in module simulusion.datasets.tinyworlds) Dreamer (class in simulusion.models.dreamer) DreamerAdapter (class in simulusion.benchmarks.adapters) DreamerAgent (class in simulusion.models.dreamer) DreamerConfig (class in simulusion.configs.dreamer_config) DreamerPolicyExport (class in simulusion.export) DreamerStepModule (class in simulusion.inference.steppers) DreamerStepper (class in simulusion.inference.steppers) DreamerV1Adapter (class in simulusion.benchmarks.adapters) DreamerV2Adapter (class in simulusion.benchmarks.adapters) DreamerWorldModel (class in simulusion.world_model.dreamer) DriftReport (class in simulusion.inference.benchmark) DROP (simulusion.configs.dit_config.DiTConfig attribute) drop_path() (in module simulusion.models.vit) drop_path_rate (simulusion.configs.genie_config.DynamicsModelConfig attribute) (simulusion.configs.genie_config.STTransformerConfig attribute) drop_rate (simulusion.configs.genie_config.DynamicsModelConfig attribute) (simulusion.configs.genie_config.STTransformerConfig attribute) DropPath (class in simulusion.blocks.st_transformer) (class in simulusion.models.vit) dump_run_config() (in module simulusion.training.run) dump_scalars_to_pickle() (simulusion.utils.dreamer_utils.Logger method) dynamics_depth (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) dynamics_dim (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) dynamics_num_heads (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) DynamicsModel (class in simulusion.models.dynamics_model) DynamicsModelConfig (class in simulusion.configs.genie_config) E early_stopping (simulusion.configs.diamond_config.DiamondConfig attribute) EARLY_STOPPING (simulusion.configs.dit_config.DiTConfig attribute) early_stopping (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) (simulusion.configs.jepa_config.JEPAConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) early_stopping_metric (simulusion.training.trainer.TrainerConfig attribute) early_stopping_mode (simulusion.training.trainer.TrainerConfig attribute) early_stopping_patience (simulusion.configs.wm_config.WMMDNRNNConfig attribute) (simulusion.configs.wm_config.WMVAEConfig attribute) EarlyStopping (class in simulusion.utils.train_utils) EDMPreconditioner (class in simulusion.models.diffusion.diamond_diffusion) ego_future (simulusion.datasets.nuplan.NuPlanSample attribute) ego_past (simulusion.datasets.nuplan.NuPlanSample attribute) elites (simulusion.agents.planning.CEMPlanner attribute) EMA (simulusion.configs.dit_config.DiTConfig attribute) ema (simulusion.configs.jepa_config.JEPAConfig attribute) EMA_DECAY (simulusion.configs.dit_config.DiTConfig attribute) ema_decay (simulusion.configs.genie_config.VideoTokenizerConfig attribute) embed_size (simulusion.models.modular_rssm.EncoderBase attribute) (simulusion.models.modular_rssm.ModularRSSM property) embedding_dim (simulusion.configs.genie_config.DynamicsModelConfig attribute) (simulusion.configs.genie_config.LatentActionModelConfig attribute) (simulusion.configs.genie_config.VideoTokenizerConfig attribute) embedding_size (simulusion.models.modular_rssm.GRUBackbone property) (simulusion.models.modular_rssm.LSTMBackbone property) (simulusion.models.modular_rssm.TransformerBackbone property) enable_console_metrics (simulusion.configs.dreamer_config.DreamerConfig attribute) enable_jsonl (simulusion.configs.dreamer_config.DreamerConfig attribute) enable_mixed_precision() (in module simulusion.utils.memory_utils) enable_performance_defaults() (in module simulusion.utils.memory_utils) enable_sweep (simulusion.configs.jepa_config.JEPAConfig attribute) enable_tensorboard (simulusion.configs.dreamer_config.DreamerConfig attribute) enable_wandb (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.jepa_config.JEPAConfig attribute) enabled (simulusion.training.precision.MixedPrecision property) enc_mask_scale (simulusion.configs.jepa_config.JEPAConfig attribute) encode() (simulusion.models.latent_action_model.LatentActionModel method) (simulusion.utils.dreamer_utils.TwoHotEncoder method) (simulusion.vision.iris_decoder.DiscreteAutoencoder method) (simulusion.vision.video_tokenizer.VideoTokenizer method) (simulusion.world_model.protocol.WorldModel method) (simulusion.world_model.protocol.WorldModelBase method) (simulusion.world_model.WorldModel method) (simulusion.world_model.WorldModelBase method) encode_to_indices() (simulusion.vision.iris_encoder.IRISEncoder method) encode_video() (simulusion.models.genie.Genie method) encoder_channels (simulusion.configs.iris_config.IRISConfig attribute) encoder_depth (simulusion.configs.genie_config.LatentActionModelConfig attribute) (simulusion.configs.genie_config.VideoTokenizerConfig attribute) encoder_dim (simulusion.configs.genie_config.LatentActionModelConfig attribute) (simulusion.configs.genie_config.VideoTokenizerConfig attribute) encoder_layers (simulusion.configs.iris_config.IRISConfig attribute) encoder_residual_blocks (simulusion.configs.iris_config.IRISConfig attribute) EncoderBase (class in simulusion.models.modular_rssm) ensure_results_dir_exists() (in module simulusion.utils.utils) entropy() (simulusion.vision.dreamer_decoder.SampleDist method) entropy_coef (simulusion.configs.iris_config.IRISConfig attribute) entropy_weight (simulusion.configs.diamond_config.DiamondConfig attribute) env (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) env_action_size() (in module simulusion.training.train_world_model) env_backend (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) env_instance (simulusion.configs.dreamer_config.DreamerConfig attribute) env_name (simulusion.configs.wm_config.WMControllerConfig attribute) env_steps_per_epoch (simulusion.configs.iris_config.IRISConfig attribute) EnvBackendSpec (class in simulusion.api) environment_steps_per_epoch (simulusion.configs.diamond_config.DiamondConfig attribute) Episode (class in simulusion.data.episode) (class in simulusion.memory.dreamer_memory) (class in simulusion.memory.planet_memory) episode_boundaries() (simulusion.datasets.diamond_dataset.ReplayBuffer method) episode_from_rlds() (in module simulusion.data.rlds) EpisodeFolder (class in simulusion.data.folder) episodes (simulusion.training.tasks.common.SequenceDataConfig attribute) EpisodeWriter (class in simulusion.data.folder) EPOCHS (simulusion.configs.dit_config.DiTConfig attribute) epochs (simulusion.configs.jepa_config.JEPAConfig attribute) epsilon_greedy (simulusion.configs.diamond_config.DiamondConfig attribute) EulerSampler (class in simulusion.models.diffusion.diamond_diffusion) eval_episodes (simulusion.configs.iris_config.IRISConfig attribute) eval_every (simulusion.training.trainer.TrainerConfig attribute) eval_interval (simulusion.configs.diamond_config.DiamondConfig attribute) eval_temperature (simulusion.configs.iris_config.IRISConfig attribute) evaluate (simulusion.configs.dreamer_config.DreamerConfig attribute) evaluate() (in module simulusion.training.train_rssm) (simulusion.benchmarks.adapters.BaseAdapter method) (simulusion.benchmarks.adapters.DiamondAdapter method) (simulusion.benchmarks.adapters.DreamerAdapter method) (simulusion.benchmarks.adapters.IRISAdapter method) (simulusion.models.dreamer.Dreamer method) (simulusion.models.dreamer.DreamerAgent method) (simulusion.training.task.TrainingTask method) (simulusion.training.train_diamond.DiamondAgent method) (simulusion.training.train_iris.IRISTrainer method) example_inputs() (simulusion.inference.steppers.DreamerStepModule method) exp_name (simulusion.configs.dreamer_config.DreamerConfig attribute) expand_state() (in module simulusion.agents.planning) expected_reward() (simulusion.models.iris_transformer.IRISTransformer method) experience_replay (simulusion.configs.dreamer_config.DreamerConfig attribute) export() (simulusion.export.ExportableAgentMixin method) (simulusion.training.task.TrainingTask method) export_any() (in module simulusion.export) export_csv() (in module simulusion.benchmarks.reporting) export_latex() (in module simulusion.benchmarks.reporting) export_markdown() (in module simulusion.benchmarks.reporting) export_model() (in module simulusion.export) export_world_model (simulusion.training.trainer.TrainerConfig attribute) ExportableAgentMixin (class in simulusion.export) extensions (simulusion.datasets.video_datasets.VideoDatasetConfig attribute) extra_repr() (simulusion.inference.quantize.Int8WeightOnlyLinear method) F factory_path (simulusion.api.EnvBackendSpec attribute) family (simulusion.world_model.diamond.DiamondWorldModel attribute) (simulusion.world_model.dreamer.DreamerWorldModel attribute) (simulusion.world_model.genie.GenieWorldModel attribute) (simulusion.world_model.iris.IRISTokenWorldModel attribute) (simulusion.world_model.planet.PlanetWorldModel attribute) (simulusion.world_model.protocol.WorldModelBase attribute) (simulusion.world_model.protocol.WorldModelSpec attribute) (simulusion.world_model.vae_mdrnn.MDRNNWorldModel attribute) (simulusion.world_model.WorldModelBase attribute) (simulusion.world_model.WorldModelSpec attribute) features() (simulusion.inference.steppers.DreamerStepper static method) files (simulusion.data.folder.EpisodeFolder attribute) filter_dataset_() (simulusion.datasets.imagenet1k.ImageNetSubset method) final_evaluation() (simulusion.training.task.TrainingTask method) final_lr (simulusion.configs.jepa_config.JEPAConfig attribute) final_max_abs_error (simulusion.inference.benchmark.DriftReport property) final_weight_decay (simulusion.configs.jepa_config.JEPAConfig attribute) FinalLayer (class in simulusion.models.diffusion.DiT) fit() (simulusion.models.diffusion.DiT.DiT class method) (simulusion.training.trainer.Trainer method) fix_init_weight() (simulusion.models.vit.VisionTransformer method) (simulusion.models.vit.VisionTransformerPredictor method) flatten_dict() (in module simulusion.utils.utils) flush() (simulusion.utils.dreamer_utils.Logger method) (simulusion.utils.logging_utils.MetricsLogger method) folder (simulusion.configs.jepa_config.JEPAConfig attribute) formats (simulusion.inference.bundle.Bundle property) forward() (simulusion.blocks.mhsa.MultiHeadSelfAttention method) (simulusion.blocks.st_transformer.DropPath method) (simulusion.blocks.st_transformer.STMLP method) (simulusion.blocks.st_transformer.STSpatialAttention method) (simulusion.blocks.st_transformer.STTemporalAttention method) (simulusion.blocks.st_transformer.STTransformer method) (simulusion.blocks.st_transformer.STTransformerBlock method) (simulusion.controller.iris_policy.CNNFeatureExtractor method) (simulusion.controller.iris_policy.IRISActor method) (simulusion.controller.iris_policy.IRISCritic method) (simulusion.controller.iris_policy.IRISPolicy method) (simulusion.export.DreamerPolicyExport method) (simulusion.export.IRISActorCriticExport method) (simulusion.inference.quantize.Int8WeightOnlyLinear method) (simulusion.inference.runtime.InferenceModel method) (simulusion.inference.steppers.DreamerStepModule method) (simulusion.models.controller.Controller method) (simulusion.models.diffusion.actor_critic.ActorCriticNetwork method) (simulusion.models.diffusion.diamond_diffusion.AdaptiveGroupNorm method) (simulusion.models.diffusion.diamond_diffusion.AttentionBlock method) (simulusion.models.diffusion.diamond_diffusion.DiffusionUNet method) (simulusion.models.diffusion.diamond_diffusion.DownBlock method) (simulusion.models.diffusion.diamond_diffusion.ResBlock method) (simulusion.models.diffusion.diamond_diffusion.TimestepEmbedding method) (simulusion.models.diffusion.diamond_diffusion.UpBlock method) (simulusion.models.diffusion.DiT.DiT method) (simulusion.models.diffusion.DiT.FinalLayer method) (simulusion.models.diffusion.DiT.LabelEmbedder method) (simulusion.models.diffusion.DiT.PatchEmbed method) (simulusion.models.diffusion.DiT.PatchUnEmbed method) (simulusion.models.diffusion.DiT.TimestepEmbedder method) (simulusion.models.diffusion.DiT.TransformerBlock method) (simulusion.models.diffusion.reward_termination.ResidualBlock method) (simulusion.models.diffusion.reward_termination.RewardTerminationLoss method) (simulusion.models.diffusion.reward_termination.RewardTerminationModel method) (simulusion.models.dreamer_rssm.RSSM method) (simulusion.models.dynamics_model.DynamicsModel method) (simulusion.models.genie.Genie method) (simulusion.models.iris_transformer.CausalSelfAttention method) (simulusion.models.iris_transformer.GPTBlock method) (simulusion.models.iris_transformer.IRISTransformer method) (simulusion.models.iris_transformer.IRISWorldModel method) (simulusion.models.latent_action_model.LatentActionModel method) (simulusion.models.mdrnn.MDRNN method) (simulusion.models.mdrnn.MDRNNCell method) (simulusion.models.modular_rssm.BackboneBase method) (simulusion.models.modular_rssm.ConvDecoder method) (simulusion.models.modular_rssm.ConvEncoder method) (simulusion.models.modular_rssm.DecoderBase method) (simulusion.models.modular_rssm.EncoderBase method) (simulusion.models.modular_rssm.GRUBackbone method) (simulusion.models.modular_rssm.LSTMBackbone method) (simulusion.models.modular_rssm.MLPDecoder method) (simulusion.models.modular_rssm.MLPEncoder method) (simulusion.models.modular_rssm.TransformerBackbone method) (simulusion.models.modular_rssm.TransformerBlock method) (simulusion.models.modular_rssm.ViTEncoder method) (simulusion.models.rssm.RecurrentStateSpaceModel method) (simulusion.models.vit.Attention method) (simulusion.models.vit.Block method) (simulusion.models.vit.ConvEmbed method) (simulusion.models.vit.DropPath method) (simulusion.models.vit.MLP method) (simulusion.models.vit.PatchEmbed method) (simulusion.models.vit.VisionTransformer method) (simulusion.models.vit.VisionTransformerPredictor method) (simulusion.training.rl_harness.ActorCritic method) (simulusion.utils.jepa_utils.AllGather static method) (simulusion.utils.jepa_utils.AllReduce static method) (simulusion.utils.jepa_utils.AllReduceSum static method) (simulusion.vision.dreamer_decoder.ActionDecoder method) (simulusion.vision.dreamer_decoder.ConvDecoder method) (simulusion.vision.dreamer_decoder.DenseDecoder method) (simulusion.vision.dreamer_encoder.ConvEncoder method) (simulusion.vision.iris_decoder.DiscreteAutoencoder method) (simulusion.vision.iris_decoder.IRISDecoder method) (simulusion.vision.iris_decoder.ResidualBlock method) (simulusion.vision.iris_decoder.UpsampleBlock method) (simulusion.vision.iris_encoder.IRISEncoder method) (simulusion.vision.iris_encoder.ResidualBlock method) (simulusion.vision.iris_encoder.SelfAttentionBlock method) (simulusion.vision.planet_decoder.CNNDecoder method) (simulusion.vision.planet_encoder.CNNEncoder method) (simulusion.vision.VAE.ConvVAE.ConvVAE method) (simulusion.vision.VAE.ConvVAE.ConvVAEDecoder method) (simulusion.vision.VAE.ConvVAE.ConvVAEEncoder method) (simulusion.vision.video_tokenizer.VideoTokenizer method) (simulusion.vision.vq_layer.VectorQuantizer method) (simulusion.vision.vq_layer.VectorQuantizerEMA method) forward_actor_critic() (simulusion.models.iris_agent.IRISAgent method) forward_cached() (simulusion.models.dynamics_model.DynamicsModel method) forward_with_cfg() (simulusion.models.diffusion.DiT.DiT method) fps (simulusion.data.folder.DatasetMeta attribute) frame_channels (simulusion.configs.iris_config.IRISConfig attribute) frame_height (simulusion.configs.iris_config.IRISConfig attribute) frame_stack (simulusion.configs.dreamer_config.DreamerConfig attribute) frame_width (simulusion.configs.iris_config.IRISConfig attribute) frames (simulusion.agents.planning.PlanningAgent attribute) (simulusion.data.episode.Episode attribute) (simulusion.world_model.protocol.Rollout attribute) (simulusion.world_model.Rollout attribute) frames_to_float() (in module simulusion.data.sequence) frameskip (simulusion.configs.diamond_config.DiamondConfig attribute) FrameStack (class in simulusion.envs.wrappers) free_nats (simulusion.configs.dreamer_config.DreamerConfig attribute) FreezeParameters (class in simulusion.utils.dreamer_utils) from_arrays() (simulusion.data.episode.Episode class method) from_checkpoints() (simulusion.world_model.vae_mdrnn.MDRNNWorldModel class method) from_config() (simulusion.models.diffusion.DiT.DiT class method) (simulusion.models.dreamer.Dreamer class method) (simulusion.models.dreamer.DreamerAgent class method) (simulusion.models.genie.Genie class method) (simulusion.models.iris_agent.IRISAgent class method) (simulusion.models.jepa_agent.JEPAAgent class method) (simulusion.training.train_diamond.DiamondAgent class method) from_dict() (simulusion.configs.jepa_config.JEPAConfig class method) from_json() (simulusion.data.folder.DatasetMeta class method) from_linear() (simulusion.inference.quantize.Int8WeightOnlyLinear class method) from_pretrained() (simulusion.models.diffusion.DiT.DiT class method) (simulusion.models.dreamer.Dreamer class method) (simulusion.models.dreamer.DreamerAgent class method) (simulusion.models.genie.Genie class method) (simulusion.models.iris_agent.IRISAgent class method) (simulusion.models.jepa_agent.JEPAAgent class method) (simulusion.training.train_diamond.DiamondAgent class method) from_use_amp() (simulusion.training.precision.MixedPrecision class method) G game (simulusion.configs.diamond_config.DiamondConfig attribute) GaussianBlur (class in simulusion.transforms.image) generate() (simulusion.models.genie.Genie method) generate_frame_cached() (simulusion.models.iris_transformer.IRISTransformer method) generate_rollouts() (in module simulusion.training.train_world_model) Genie (class in simulusion.models.genie) genie_kwargs_from_config() (in module simulusion.models.genie) GenieConfig (class in simulusion.configs.genie_config) (class in simulusion.training.train_genie) GenieSmallConfig (class in simulusion.configs.genie_config) GenieTrainer (class in simulusion.training.train_genie) GenieWorldModel (class in simulusion.world_model.genie) get_1d_sincos_pos_embed() (in module simulusion.models.vit) get_1d_sincos_pos_embed_from_grid() (in module simulusion.models.vit) get_2d_sincos_pos_embed() (in module simulusion.models.diffusion.DiT) (in module simulusion.models.vit) get_2d_sincos_pos_embed_from_grid() (in module simulusion.models.vit) get_action() (simulusion.controller.iris_policy.IRISActor method) (simulusion.models.diffusion.actor_critic.ActorCriticNetwork method) (simulusion.training.rl_harness.ActorCritic method) get_actions() (simulusion.models.diffusion.actor_critic.ActorCriticNetwork method) get_arrays() (simulusion.memory.iris_memory.IRISOnPolicyBuffer method) get_autoencoder_config() (simulusion.configs.iris_config.IRISConfig method) get_available_memory() (in module simulusion.models.dreamer) get_combined_params() (in module simulusion.utils.utils) get_dataset_info() (simulusion.datasets.tinyworlds.TinyWorldsDataLoader static method) get_default_device() (in module simulusion.configs.diamond_config) get_dist() (simulusion.models.dreamer_rssm.RSSM method) (simulusion.models.modular_rssm.ModularRSSM method) get_dit_config() (in module simulusion.configs.dit_config) get_embed_size() (simulusion.models.modular_rssm.EncoderBase method) get_env_backend_spec() (in module simulusion.api) get_epsilon() (simulusion.training.train_iris.IRISTrainer method) get_frame_shape() (simulusion.configs.iris_config.IRISConfig method) get_hidden_size() (simulusion.models.diffusion.actor_critic.ActorCriticNetwork method) get_info() (simulusion.datasets.tinyworlds.TinyWorldsDataset method) get_init_hidden() (simulusion.models.mdrnn.MDRNN method) (simulusion.models.mdrnn.MDRNNCell method) get_init_state() (simulusion.models.rssm.RecurrentStateSpaceModel method) get_intermediate_layers() (simulusion.models.vit.VisionTransformer method) get_loss() (in module simulusion.training.train_mdn_rnn) get_mask() (in module simulusion.utils.utils) get_mask_prob() (simulusion.models.dynamics_model.MaskGITSampler method) get_model_spec() (in module simulusion.api) get_num_parameters() (simulusion.models.genie.Genie method) get_package_logger() (in module simulusion.utils.logging_utils) get_parameters() (in module simulusion.utils.dreamer_utils) get_posterior() (simulusion.models.dreamer_rssm.RSSM method) get_preconditioners() (simulusion.models.diffusion.diamond_diffusion.EDMPreconditioner method) get_prior() (simulusion.models.dreamer_rssm.RSSM method) get_rl_config() (simulusion.configs.iris_config.IRISConfig method) get_task() (in module simulusion.training.task) get_transformer_config() (simulusion.configs.iris_config.IRISConfig method) get_value() (simulusion.models.diffusion.actor_critic.ActorCriticNetwork method) gmm_components (simulusion.configs.wm_config.WMMDNRNNConfig attribute) gmm_loss() (in module simulusion.losses.gmm_loss) gpt_blocks() (simulusion.models.iris_transformer.IRISTransformer method) GPTBlock (class in simulusion.models.iris_transformer) gpu_timer() (in module simulusion.utils.jepa_utils) grad_accumulation (simulusion.training.trainer.TrainerConfig attribute) grad_clip_norm (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) grad_logger() (in module simulusion.utils.jepa_utils) grad_scaler() (simulusion.training.precision.MixedPrecision method) gradient_checkpointing (simulusion.configs.iris_config.IRISConfig attribute) gradient_update() (simulusion.training.trainer.Trainer method) GradientUpdate (class in simulusion.training.precision) GraphedDecodeStep (class in simulusion.models.iris_transformer) GRUBackbone (class in simulusion.models.modular_rssm) gym_render_mode (simulusion.configs.dreamer_config.DreamerConfig attribute) GymImageEnv (class in simulusion.envs.gym_env) H has_actions (simulusion.data.episode.Episode property) has_reward (simulusion.world_model.protocol.WorldModelSpec attribute) (simulusion.world_model.WorldModelSpec attribute) has_termination (simulusion.world_model.protocol.WorldModelSpec attribute) (simulusion.world_model.WorldModelSpec attribute) HDF5Dataset (class in simulusion.datasets.video_datasets) HEADS (simulusion.configs.dit_config.DiTConfig attribute) height (simulusion.configs.wm_config.WMVAEConfig attribute) hidden_size (simulusion.configs.wm_config.WMControllerConfig attribute) (simulusion.configs.wm_config.WMMDNRNNConfig attribute) high (simulusion.world_model.ActionSpec attribute) (simulusion.world_model.protocol.ActionSpec attribute) horizon (simulusion.agents.planning.CEMPlanner attribute) I image (simulusion.data.rlds.RLDSKeys attribute) image_folder (simulusion.configs.jepa_config.JEPAConfig attribute) image_size (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.genie_config.DynamicsModelConfig attribute) (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.configs.genie_config.LatentActionModelConfig attribute) (simulusion.configs.genie_config.VideoTokenizerConfig attribute) (simulusion.datasets.tinyworlds.TinyWorldsConfig attribute) (simulusion.datasets.video_datasets.DatasetConfig attribute) (simulusion.training.tasks.common.SequenceDataConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) ImageFolderDataset (class in simulusion.datasets.video_datasets) ImageNet (class in simulusion.datasets.imagenet1k) ImageNetSubset (class in simulusion.datasets.imagenet1k) imagination_horizon (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) imagine() (simulusion.inference.steppers.DreamerStepper method) (simulusion.models.iris_transformer.IRISWorldModel method) imagine_horizon (simulusion.configs.dreamer_config.DreamerConfig attribute) imagine_rollout() (simulusion.models.dreamer_rssm.RSSM method) (simulusion.models.iris_agent.IRISAgent method) (simulusion.models.modular_rssm.ModularRSSM method) imagine_step() (simulusion.models.dreamer_rssm.RSSM method) (simulusion.models.iris_transformer.IRISTransformer method) (simulusion.models.modular_rssm.ModularRSSM method) ImagineOutput (class in simulusion.inference.steppers) IMG_SIZE (simulusion.configs.dit_config.DiTConfig attribute) import_lerobot() (in module simulusion.data.lerobot) import_path (simulusion.api.ModelSpec attribute) in_channels (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.configs.genie_config.LatentActionModelConfig attribute) (simulusion.configs.genie_config.VideoTokenizerConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) include_reward (simulusion.configs.wm_config.WMMDNRNNConfig attribute) infer() (simulusion.data.folder.DatasetMeta class method) infer_actions() (simulusion.models.genie.Genie method) inference_context() (in module simulusion.inference.precision) InferenceModel (class in simulusion.inference.runtime) infinite() (simulusion.training.trainer.Trainer static method) info (simulusion.data.folder.DatasetMeta attribute) (simulusion.world_model.protocol.Transition attribute) (simulusion.world_model.Transition attribute) init_cache() (simulusion.blocks.st_transformer.STTransformer method) (simulusion.models.dynamics_model.DynamicsModel method) (simulusion.models.iris_transformer.IRISTransformer method) init_distributed() (in module simulusion.utils.jepa_utils) init_hidden() (simulusion.controller.iris_policy.IRISPolicy method) (simulusion.models.diffusion.actor_critic.ActorCriticNetwork method) (simulusion.models.diffusion.reward_termination.RewardTerminationModel method) init_hidden_state() (simulusion.controller.iris_policy.IRISActor method) (simulusion.controller.iris_policy.IRISCritic method) init_model() (in module simulusion.helpers.jepa_helper) init_opt() (in module simulusion.helpers.jepa_helper) init_state() (simulusion.inference.steppers.DreamerStepper method) (simulusion.inference.steppers.IRISStepper method) (simulusion.inference.steppers.WorldModelStepper method) (simulusion.models.dreamer_rssm.RSSM method) (simulusion.models.modular_rssm.BackboneBase method) (simulusion.models.modular_rssm.GRUBackbone method) (simulusion.models.modular_rssm.LSTMBackbone method) (simulusion.models.modular_rssm.ModularRSSM method) (simulusion.models.modular_rssm.TransformerBackbone method) init_std (simulusion.agents.planning.CEMPlanner attribute) Int8WeightOnlyLinear (class in simulusion.inference.quantize) interpolate_pos_encoding() (simulusion.models.vit.VisionTransformer method) ipe_scale (simulusion.configs.jepa_config.JEPAConfig attribute) iqm_of_array() (in module simulusion.benchmarks.metrics) IRISActor (class in simulusion.controller.iris_policy) IRISActorCriticExport (class in simulusion.export) IRISAdapter (class in simulusion.benchmarks.adapters) IRISAgent (class in simulusion.models.iris_agent) IRISConfig (class in simulusion.configs.iris_config) IRISCritic (class in simulusion.controller.iris_policy) IRISDecoder (class in simulusion.vision.iris_decoder) IRISEncoder (class in simulusion.vision.iris_encoder) IRISOnPolicyBuffer (class in simulusion.memory.iris_memory) IRISPolicy (class in simulusion.controller.iris_policy) IRISReplayBuffer (class in simulusion.memory.iris_memory) IRISStepper (class in simulusion.inference.steppers) IRISTokenWorldModel (class in simulusion.world_model.iris) IRISTrainer (class in simulusion.training.train_iris) IRISTransformer (class in simulusion.models.iris_transformer) IRISWorldModel (class in simulusion.models.iris_transformer) is_moved_mujoco_error() (in module simulusion.envs.robotics_env) is_ready() (simulusion.datasets.diamond_dataset.ReplayBuffer method) items_per_sec (simulusion.inference.benchmark.LatencyReport attribute) iterations (simulusion.agents.planning.CEMPlanner attribute) (simulusion.inference.benchmark.LatencyReport attribute) (simulusion.training.trainer.TrainResult attribute) J JEPAAgent (class in simulusion.models.jepa_agent) JEPAConfig (class in simulusion.configs.jepa_config) jsonl_filename (simulusion.configs.dreamer_config.DreamerConfig attribute) K keep_checkpoints (simulusion.training.trainer.TrainerConfig attribute) kind (simulusion.world_model.ActionSpec attribute) (simulusion.world_model.protocol.ActionSpec attribute) kl_alpha (simulusion.configs.dreamer_config.DreamerConfig attribute) kl_loss_coeff (simulusion.configs.dreamer_config.DreamerConfig attribute) KVCache (class in simulusion.models.iris_transformer) L LabelEmbedder (class in simulusion.models.diffusion.DiT) lambda_returns (simulusion.configs.diamond_config.DiamondConfig attribute) language (simulusion.data.episode.Episode attribute) (simulusion.data.rlds.RLDSKeys attribute) LatencyReport (class in simulusion.inference.benchmark) latent (simulusion.world_model.ActionSpec attribute) (simulusion.world_model.protocol.ActionSpec attribute) latent_size (simulusion.configs.wm_config.WMControllerConfig attribute) (simulusion.configs.wm_config.WMMDNRNNConfig attribute) (simulusion.configs.wm_config.WMVAEConfig attribute) LatentActionModel (class in simulusion.models.latent_action_model) LatentActionModelConfig (class in simulusion.configs.genie_config) LatentSequenceDataset (class in simulusion.datasets.wm_dataset) latest_checkpoint() (in module simulusion.training.trainer) LEARN_SIGMA (simulusion.configs.dit_config.DiTConfig attribute) learning_rate (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.configs.wm_config.WMMDNRNNConfig attribute) (simulusion.configs.wm_config.WMVAEConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) lengths (simulusion.data.folder.EpisodeFolder property) (simulusion.data.lerobot.LeRobotEpisodes property) LeRobotEpisodes (class in simulusion.data.lerobot) lineplot() (in module simulusion.utils.utils) list_available_atari_envs() (in module simulusion.envs.ale_atari_env) list_available_datasets() (simulusion.datasets.tinyworlds.TinyWorldsDataLoader static method) list_dit_presets() (in module simulusion.configs.dit_config) list_env_backends() (in module simulusion.api) list_envs() (in module simulusion.api) list_gymnasium_robotics_envs() (in module simulusion.envs.robotics_env) list_models() (in module simulusion.api) list_procgen_envs() (in module simulusion.envs.procgen_env) list_tasks() (in module simulusion.training.task) list_world_model_families() (in module simulusion.world_model) load() (simulusion.data.episode.Episode class method) (simulusion.models.iris_agent.IRISAgent method) (simulusion.world_model.protocol.WorldModelBase class method) (simulusion.world_model.WorldModelBase class method) load_artifact() (simulusion.inference.bundle.Bundle method) load_bundle() (in module simulusion.inference.bundle) load_checkpoint (simulusion.configs.jepa_config.JEPAConfig attribute) load_checkpoint() (in module simulusion.helpers.jepa_helper) (simulusion.benchmarks.adapters.BaseAdapter method) (simulusion.benchmarks.adapters.DiamondAdapter method) (simulusion.benchmarks.adapters.DreamerAdapter method) (simulusion.benchmarks.adapters.IRISAdapter method) (simulusion.training.task.TrainingTask method) (simulusion.training.train_diamond.DiamondAgent method) (simulusion.training.train_genie.GenieTrainer method) load_episode() (in module simulusion.data.folder) load_exported() (in module simulusion.export) load_memory() (in module simulusion.utils.utils) load_next_buffer() (simulusion.datasets.wm_dataset.RolloutDataset method) load_replay() (in module simulusion.training.tasks.common) load_state_dict() (simulusion.datasets.diamond_dataset.ReplayBuffer method) (simulusion.training.precision.GradientUpdate method) (simulusion.training.task.TrainingTask method) (simulusion.utils.train_utils.EarlyStopping method) (simulusion.utils.train_utils.ReduceLROnPlateau method) load_weights() (simulusion.inference.bundle.Bundle method) load_world_model() (in module simulusion.world_model) load_yml_config() (in module simulusion.utils.utils) log() (simulusion.utils.jepa_utils.CSVLogger method) (simulusion.utils.logging_utils.MetricsLogger method) log_abs_det_jacobian() (simulusion.vision.dreamer_decoder.TanhBijector method) log_dir (simulusion.configs.dreamer_config.DreamerConfig attribute) log_every (simulusion.training.trainer.TrainerConfig attribute) log_file (simulusion.configs.dreamer_config.DreamerConfig attribute) log_interval (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) log_level (simulusion.configs.dreamer_config.DreamerConfig attribute) log_scalar() (simulusion.utils.dreamer_utils.Logger method) log_scalars() (simulusion.utils.dreamer_utils.Logger method) log_system_stats_freq (simulusion.configs.dreamer_config.DreamerConfig attribute) log_video() (simulusion.utils.logging_utils.MetricsLogger method) log_video_freq (simulusion.configs.dreamer_config.DreamerConfig attribute) log_videos() (simulusion.utils.dreamer_utils.Logger method) logdir (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.wm_config.WMControllerConfig attribute) (simulusion.configs.wm_config.WMMDNRNNConfig attribute) (simulusion.configs.wm_config.WMVAEConfig attribute) Logger (class in simulusion.utils.dreamer_utils) loss_type (simulusion.configs.jepa_config.JEPAConfig attribute) low (simulusion.world_model.ActionSpec attribute) (simulusion.world_model.protocol.ActionSpec attribute) LR (simulusion.configs.dit_config.DiTConfig attribute) lr (simulusion.configs.jepa_config.JEPAConfig attribute) (simulusion.utils.train_utils.ReduceLROnPlateau property) lr_reference_batch_size (simulusion.configs.jepa_config.JEPAConfig attribute) LSTMBackbone (class in simulusion.models.modular_rssm) M main() (in module simulusion.training.run) (in module simulusion.training.train_diamond) (in module simulusion.training.train_genie) (in module simulusion.training.train_iris) (in module simulusion.training.train_jepa) (in module simulusion.training.train_planet) (in module simulusion.training.train_rssm) (in module simulusion.training.train_world_model) main_from_cli() (in module simulusion.training.train_jepa) make_atari_env() (in module simulusion.envs.ale_atari_env) make_atari_vector_env() (in module simulusion.envs.ale_atari_vector_env) make_cifar10() (in module simulusion.datasets.cifar10) make_controller_env() (in module simulusion.training.train_controller) make_diamond_atari_env() (in module simulusion.envs.diamond_atari) make_dmc_env() (in module simulusion.envs.dmc) make_dmlab_env() (in module simulusion.envs.dmlab) make_env() (in module simulusion.api) (in module simulusion.models.dreamer) make_gym_env() (in module simulusion.envs.gym_env) make_gymnasium_env_with_robotics_fallback() (in module simulusion.envs.robotics_env) make_imagefolder() (in module simulusion.datasets.imagenet1k) make_imagenet1k() (in module simulusion.datasets.imagenet1k) make_jepa_loaders() (in module simulusion.training.train_jepa) make_mujoco_env() (in module simulusion.envs.mujoco_env) make_mujoco_env_from_config() (in module simulusion.envs.mujoco_env) make_nuplan_dataloader() (in module simulusion.datasets.nuplan) make_planet_scheduler() (in module simulusion.models.planet) make_procgen_env() (in module simulusion.envs.procgen_env) make_robotics_env() (in module simulusion.envs.robotics_env) make_stepper() (in module simulusion.inference.steppers) make_transforms() (in module simulusion.transforms.image) make_unity_mlagents_env() (in module simulusion.envs.unity_env) manifest (simulusion.inference.bundle.Bundle attribute) MANIFEST_NAME (in module simulusion.world_model.protocol) map_raster (simulusion.datasets.nuplan.NuPlanSample attribute) mask_prob_max (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) mask_prob_min (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) MaskCollator (class in simulusion.masks.multiblock) (class in simulusion.masks.random) maskgit_steps (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) MaskGITSampler (class in simulusion.models.dynamics_model) max_abs_error (simulusion.inference.benchmark.DriftReport attribute) max_context (simulusion.world_model.protocol.WorldModelSpec attribute) (simulusion.world_model.WorldModelSpec attribute) max_env_steps (simulusion.configs.iris_config.IRISConfig attribute) max_episode_length (simulusion.configs.dreamer_config.DreamerConfig attribute) max_episode_steps (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) (simulusion.envs.dmlab.DMLabEnv property) (simulusion.envs.gym_env.GymImageEnv property) (simulusion.envs.procgen_env.ProcgenImageEnv property) (simulusion.envs.unity_env.UnityMLAgentsEnv property) (simulusion.utils.utils.TorchImageEnvWrapper property) max_noop (simulusion.configs.diamond_config.DiamondConfig attribute) max_steps (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) (simulusion.training.trainer.TrainerConfig attribute) max_steps() (simulusion.training.task.TrainingTask method) max_videos_to_save (simulusion.configs.dreamer_config.DreamerConfig attribute) max_workers (simulusion.configs.wm_config.WMControllerConfig attribute) maybe_compile() (in module simulusion.utils.memory_utils) MDRNN (class in simulusion.models.mdrnn) MDRNNCell (class in simulusion.models.mdrnn) MDRNNWorldModel (class in simulusion.world_model.vae_mdrnn) mean() (simulusion.vision.dreamer_decoder.SampleDist method) mean_action() (simulusion.vision.dreamer_decoder.ActionDecoder method) mean_metrics() (in module simulusion.training.tasks.common) mean_ms (simulusion.inference.benchmark.LatencyReport attribute) Memory (class in simulusion.memory.dreamer_memory) (class in simulusion.memory.planet_memory) meta (simulusion.data.lerobot.LeRobotEpisodes property) metrics (simulusion.models.dreamer.Dreamer attribute) (simulusion.training.trainer.TrainResult attribute) MetricsLogger (class in simulusion.utils.logging_utils) min_delta (simulusion.configs.diamond_config.DiamondConfig attribute) MIN_DELTA (simulusion.configs.dit_config.DiTConfig attribute) min_delta (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) (simulusion.configs.jepa_config.JEPAConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) (simulusion.training.trainer.TrainerConfig attribute) min_keep (simulusion.configs.jepa_config.JEPAConfig attribute) min_ms (simulusion.inference.benchmark.LatencyReport attribute) min_std (simulusion.agents.planning.CEMPlanner attribute) MixedPrecision (class in simulusion.training.precision) MLP (class in simulusion.models.vit) mlp_ratio (simulusion.configs.genie_config.DynamicsModelConfig attribute) (simulusion.configs.genie_config.STTransformerConfig attribute) MLPDecoder (class in simulusion.models.modular_rssm) MLPEncoder (class in simulusion.models.modular_rssm) mode() (simulusion.vision.dreamer_decoder.SampleDist method) model_learning_rate (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) model_name (simulusion.configs.jepa_config.JEPAConfig attribute) model_path (simulusion.training.trainer.TrainResult attribute) ModelPreset (class in simulusion.configs.diamond_config) ModelSpec (class in simulusion.api) ModularRSSM (class in simulusion.models.modular_rssm) modulate() (in module simulusion.models.diffusion.DiT) module simulusion.agents.planning simulusion.api simulusion.benchmarks simulusion.benchmarks.adapters simulusion.benchmarks.metrics simulusion.benchmarks.reporting simulusion.benchmarks.runner simulusion.blocks simulusion.blocks.mhsa simulusion.blocks.st_transformer simulusion.catalog simulusion.configs.diamond_config simulusion.configs.dit_config simulusion.configs.dreamer_config simulusion.configs.genie_config simulusion.configs.iris_config simulusion.configs.jepa_config simulusion.configs.wm_config simulusion.controller.iris_policy simulusion.controller.rollout_generator simulusion.controller.rssm_policy simulusion.data.episode simulusion.data.folder simulusion.data.lerobot simulusion.data.rlds simulusion.data.sequence simulusion.datasets.cifar10 simulusion.datasets.diamond_dataset simulusion.datasets.imagenet1k simulusion.datasets.nuplan simulusion.datasets.tinyworlds simulusion.datasets.video_datasets simulusion.datasets.wm_dataset simulusion.envs.ale_atari_env simulusion.envs.ale_atari_vector_env simulusion.envs.diamond_atari simulusion.envs.dmc simulusion.envs.dmlab simulusion.envs.gym_env simulusion.envs.mujoco_env simulusion.envs.procgen_env simulusion.envs.robotics_env simulusion.envs.unity_env simulusion.envs.vector_env simulusion.envs.wrappers simulusion.export simulusion.helpers.jepa_helper simulusion.inference.benchmark simulusion.inference.bundle simulusion.inference.precision simulusion.inference.quantize simulusion.inference.runtime simulusion.inference.steppers simulusion.losses.convae_loss simulusion.losses.gmm_loss simulusion.masks simulusion.masks.default simulusion.masks.multiblock simulusion.masks.random simulusion.memory.dreamer_memory simulusion.memory.iris_memory simulusion.memory.planet_memory simulusion.models.controller simulusion.models.diffusion simulusion.models.diffusion.actor_critic simulusion.models.diffusion.DDPM simulusion.models.diffusion.diamond_diffusion simulusion.models.diffusion.DiT simulusion.models.diffusion.reward_termination simulusion.models.dreamer simulusion.models.dreamer_rssm simulusion.models.dynamics_model simulusion.models.genie simulusion.models.iris_agent simulusion.models.iris_transformer simulusion.models.jepa_agent simulusion.models.latent_action_model simulusion.models.mdrnn simulusion.models.modular_rssm simulusion.models.planet simulusion.models.rssm simulusion.models.vit simulusion.training simulusion.training.precision simulusion.training.rl_harness simulusion.training.run simulusion.training.task simulusion.training.tasks.common simulusion.training.train_controller simulusion.training.train_convvae simulusion.training.train_diamond simulusion.training.train_genie simulusion.training.train_iris simulusion.training.train_jepa simulusion.training.train_mdn_rnn simulusion.training.train_planet simulusion.training.train_rssm simulusion.training.train_world_model simulusion.training.trainer simulusion.transforms.image simulusion.utils.data_utils simulusion.utils.dreamer_utils simulusion.utils.jepa_utils simulusion.utils.logging_utils simulusion.utils.memory_utils simulusion.utils.train_utils simulusion.utils.utils simulusion.vision.dreamer_decoder simulusion.vision.dreamer_encoder simulusion.vision.iris_decoder simulusion.vision.iris_encoder simulusion.vision.planet_decoder simulusion.vision.planet_encoder simulusion.vision.VAE.ConvVAE simulusion.vision.video_tokenizer simulusion.vision.vq_layer simulusion.world_model simulusion.world_model.diamond simulusion.world_model.dreamer simulusion.world_model.genie simulusion.world_model.iris simulusion.world_model.planet simulusion.world_model.protocol simulusion.world_model.vae_mdrnn Mp4Writer (class in simulusion.utils.utils) mujoco_binary_path (simulusion.configs.dreamer_config.DreamerConfig attribute) mujoco_camera (simulusion.configs.dreamer_config.DreamerConfig attribute) mujoco_frame_skip (simulusion.configs.dreamer_config.DreamerConfig attribute) mujoco_reset_noise_scale (simulusion.configs.dreamer_config.DreamerConfig attribute) mujoco_xml_path (simulusion.configs.dreamer_config.DreamerConfig attribute) mujoco_xml_string (simulusion.configs.dreamer_config.DreamerConfig attribute) MuJoCoImageEnv (class in simulusion.envs.mujoco_env) MultiAgentBenchmarkRunner (class in simulusion.benchmarks.runner) MultiblockMaskCollator (in module simulusion.masks) MultiHeadSelfAttention (class in simulusion.blocks.mhsa) N n (simulusion.world_model.ActionSpec attribute) (simulusion.world_model.protocol.ActionSpec attribute) n_samples (simulusion.configs.wm_config.WMControllerConfig attribute) name (simulusion.api.EnvBackendSpec attribute) (simulusion.api.ModelSpec attribute) (simulusion.inference.benchmark.LatencyReport attribute) (simulusion.training.task.TrainingTask attribute) (simulusion.vision.dreamer_decoder.SampleDist property) no_gpu (simulusion.configs.dreamer_config.DreamerConfig attribute) noreload (simulusion.configs.wm_config.WMMDNRNNConfig attribute) (simulusion.configs.wm_config.WMVAEConfig attribute) normalize_frames_for_saving() (in module simulusion.utils.utils) normalize_procgen_env_name() (in module simulusion.envs.procgen_env) NormalizeActions (class in simulusion.envs.wrappers) nosamples (simulusion.configs.wm_config.WMVAEConfig attribute) npz_length() (in module simulusion.data.episode) num_buckets (simulusion.configs.dreamer_config.DreamerConfig attribute) NUM_CLASSES (simulusion.configs.dit_config.DiTConfig attribute) num_conditioning_frames (simulusion.configs.diamond_config.DiamondConfig attribute) num_enc_masks (simulusion.configs.jepa_config.JEPAConfig attribute) num_epochs (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.wm_config.WMMDNRNNConfig attribute) (simulusion.configs.wm_config.WMVAEConfig attribute) num_frames (simulusion.configs.genie_config.DynamicsModelConfig attribute) (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.configs.genie_config.LatentActionModelConfig attribute) (simulusion.configs.genie_config.STTransformerConfig attribute) (simulusion.configs.genie_config.VideoTokenizerConfig attribute) (simulusion.datasets.tinyworlds.TinyWorldsConfig attribute) (simulusion.datasets.video_datasets.DatasetConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) num_heads (simulusion.configs.genie_config.DynamicsModelConfig attribute) (simulusion.configs.genie_config.LatentActionModelConfig attribute) (simulusion.configs.genie_config.STTransformerConfig attribute) (simulusion.configs.genie_config.VideoTokenizerConfig attribute) num_patches_per_frame (simulusion.configs.genie_config.STTransformerConfig attribute) num_pred_masks (simulusion.configs.jepa_config.JEPAConfig attribute) num_sampling_steps (simulusion.configs.diamond_config.DiamondConfig attribute) num_transitions (simulusion.data.episode.Episode property) num_units (simulusion.configs.dreamer_config.DreamerConfig attribute) NUM_WORKERS (simulusion.configs.dit_config.DiTConfig attribute) num_workers (simulusion.configs.jepa_config.JEPAConfig attribute) (simulusion.datasets.tinyworlds.TinyWorldsConfig attribute) (simulusion.datasets.video_datasets.DatasetConfig attribute) (simulusion.training.tasks.common.SequenceDataConfig attribute) NumPyDataset (class in simulusion.datasets.video_datasets) NuPlanDataset (class in simulusion.datasets.nuplan) NuPlanSample (class in simulusion.datasets.nuplan) O obs_embed_size (simulusion.configs.dreamer_config.DreamerConfig attribute) obs_key (simulusion.datasets.video_datasets.VideoDatasetConfig attribute) obs_shape (simulusion.world_model.protocol.WorldModelSpec attribute) (simulusion.world_model.WorldModelSpec attribute) obs_size (simulusion.configs.diamond_config.DiamondConfig attribute) obs_space (simulusion.envs.wrappers.RenderImage property) (simulusion.envs.wrappers.ResizeImage property) ObsDict (class in simulusion.envs.wrappers) observation_keys (simulusion.data.folder.DatasetMeta attribute) observation_size (simulusion.utils.utils.TorchImageEnvWrapper property) observation_space (simulusion.envs.dmc.DeepMindControlEnv property) (simulusion.envs.dmlab.DMLabEnv property) (simulusion.envs.gym_env.GymImageEnv property) (simulusion.envs.mujoco_env.MuJoCoImageEnv property) (simulusion.envs.procgen_env.ProcgenImageEnv property) (simulusion.envs.unity_env.UnityMLAgentsEnv property) (simulusion.envs.wrappers.FrameStack property) (simulusion.envs.wrappers.ObsDict property) (simulusion.envs.wrappers.RewardObs property) ObservationDataset (class in simulusion.datasets.wm_dataset) observations (simulusion.data.episode.Episode attribute) observe() (simulusion.agents.planning.PlanningAgent method) (simulusion.inference.steppers.DreamerStepper method) (simulusion.inference.steppers.IRISStepper method) (simulusion.inference.steppers.WorldModelStepper method) observe_rollout() (simulusion.models.dreamer_rssm.RSSM method) (simulusion.models.modular_rssm.ModularRSSM method) observe_step() (simulusion.models.dreamer_rssm.RSSM method) (simulusion.models.modular_rssm.ModularRSSM method) one_hot_actions() (in module simulusion.data.sequence) OneHotAction (class in simulusion.envs.wrappers) open_episodes() (in module simulusion.training.tasks.common) optimize_for_inference() (in module simulusion.inference.runtime) optimize_memory_efficient_ops() (in module simulusion.utils.memory_utils) out_dir (simulusion.training.trainer.TrainerConfig attribute) P p50_ms (simulusion.inference.benchmark.LatencyReport attribute) p90_ms (simulusion.inference.benchmark.LatencyReport attribute) p99_ms (simulusion.inference.benchmark.LatencyReport attribute) p_mean (simulusion.configs.diamond_config.DiamondConfig attribute) p_sample() (simulusion.models.diffusion.DDPM.DDPM method) p_std (simulusion.configs.diamond_config.DiamondConfig attribute) parameter_count() (simulusion.models.diffusion.DiT.DiT method) (simulusion.models.dreamer.Dreamer method) (simulusion.models.dreamer.DreamerAgent method) (simulusion.models.genie.Genie method) (simulusion.models.iris_agent.IRISAgent method) (simulusion.models.jepa_agent.JEPAAgent method) (simulusion.training.train_diamond.DiamondAgent method) PATCH (simulusion.configs.dit_config.DiTConfig attribute) patch_size (simulusion.configs.genie_config.DynamicsModelConfig attribute) (simulusion.configs.genie_config.LatentActionModelConfig attribute) (simulusion.configs.genie_config.VideoTokenizerConfig attribute) (simulusion.configs.jepa_config.JEPAConfig attribute) PatchEmbed (class in simulusion.models.diffusion.DiT) (class in simulusion.models.vit) PatchUnEmbed (class in simulusion.models.diffusion.DiT) path (simulusion.inference.bundle.Bundle attribute) (simulusion.training.tasks.common.SequenceDataConfig attribute) patience (simulusion.configs.diamond_config.DiamondConfig attribute) PATIENCE (simulusion.configs.dit_config.DiTConfig attribute) patience (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) (simulusion.configs.jepa_config.JEPAConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) (simulusion.training.trainer.TrainerConfig attribute) peak_memory_mb (simulusion.inference.benchmark.LatencyReport attribute) perceptual_blocks (simulusion.configs.iris_config.IRISConfig attribute) perceptual_linear_weights (simulusion.configs.iris_config.IRISConfig attribute) perceptual_weight (simulusion.configs.iris_config.IRISConfig attribute) perf_defaults (simulusion.configs.dreamer_config.DreamerConfig attribute) persistent_workers (simulusion.configs.diamond_config.DiamondConfig attribute) pin_mem (simulusion.configs.jepa_config.JEPAConfig attribute) pin_memory (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.datasets.video_datasets.DatasetConfig attribute) plan() (simulusion.agents.planning.CEMPlanner method) Planet (class in simulusion.models.planet) PlanetWorldModel (class in simulusion.world_model.planet) planning_target (simulusion.datasets.nuplan.NuPlanSample attribute) PlanningAgent (class in simulusion.agents.planning) play() (simulusion.models.genie.Genie method) plot_metrics() (in module simulusion.utils.utils) policy_loss() (simulusion.models.diffusion.actor_critic.RLLoss method) poll() (simulusion.controller.rssm_policy.RSSMPolicy method) pop_size (simulusion.configs.wm_config.WMControllerConfig attribute) postprocess_img() (in module simulusion.utils.utils) PPOTrainer (class in simulusion.training.rl_harness) precision (simulusion.training.trainer.TrainerConfig attribute) precision_dtype() (in module simulusion.inference.precision) pred_depth (simulusion.configs.jepa_config.JEPAConfig attribute) pred_emb_dim (simulusion.configs.jepa_config.JEPAConfig attribute) pred_mask_scale (simulusion.configs.jepa_config.JEPAConfig attribute) pred_reward() (simulusion.models.rssm.RecurrentStateSpaceModel method) predict() (simulusion.models.diffusion.reward_termination.RewardTerminationModel method) predict_next_tokens() (simulusion.models.iris_transformer.IRISTransformer method) prefetch_iterator() (in module simulusion.utils.data_utils) preload (simulusion.training.tasks.common.SequenceDataConfig attribute) prepare_dataloader() (simulusion.training.trainer.Trainer method) prepare_module() (simulusion.training.trainer.Trainer method) preprocess_frame() (simulusion.training.train_iris.IRISTrainer method) preprocess_img() (in module simulusion.utils.utils) preprocess_obs() (in module simulusion.models.dreamer) preset (simulusion.configs.diamond_config.DiamondConfig attribute) prime_cache() (simulusion.models.iris_transformer.IRISTransformer method) procgen_distribution_mode (simulusion.configs.dreamer_config.DreamerConfig attribute) procgen_num_levels (simulusion.configs.dreamer_config.DreamerConfig attribute) procgen_start_level (simulusion.configs.dreamer_config.DreamerConfig attribute) ProcgenImageEnv (class in simulusion.envs.procgen_env) progress (simulusion.training.trainer.TrainResult attribute) progress() (simulusion.training.task.TrainingTask method) progress_unit (simulusion.training.task.TrainingTask attribute) Q q_sample() (simulusion.models.diffusion.DDPM.DDPM method) qkv_bias (simulusion.configs.genie_config.DynamicsModelConfig attribute) (simulusion.configs.genie_config.STTransformerConfig attribute) quantize_weights() (in module simulusion.inference.quantize) quantizer (simulusion.configs.iris_config.IRISConfig attribute) R RandomMaskCollator (in module simulusion.masks) read_checkpoint (simulusion.configs.jepa_config.JEPAConfig attribute) read_manifest() (in module simulusion.world_model.protocol) read_meta() (in module simulusion.data.folder) reconstruct() (simulusion.models.iris_agent.IRISAgent method) reconstruct_chunk_size (simulusion.models.iris_agent.IRISAgent attribute) reconstruction_weight (simulusion.configs.iris_config.IRISConfig attribute) RecurrentStateSpaceModel (class in simulusion.models.rssm) recursive (simulusion.datasets.video_datasets.VideoDatasetConfig attribute) ReduceLROnPlateau (class in simulusion.utils.train_utils) register_buffers() (simulusion.utils.dreamer_utils.TwoHotEncoder method) register_gymnasium_robotics_envs() (in module simulusion.envs.robotics_env) register_task() (in module simulusion.training.task) rel_error (simulusion.inference.benchmark.DriftReport attribute) release() (simulusion.utils.utils.Mp4Writer method) render (simulusion.configs.dreamer_config.DreamerConfig attribute) render() (simulusion.envs.diamond_atari.DiamondAtariWrapper method) (simulusion.envs.dmc.DeepMindControlEnv method) (simulusion.envs.dmlab.DMLabEnv method) (simulusion.envs.gym_env.GymImageEnv method) (simulusion.envs.mujoco_env.MuJoCoImageEnv method) (simulusion.envs.procgen_env.ProcgenImageEnv method) (simulusion.envs.unity_env.UnityMLAgentsEnv method) (simulusion.utils.utils.TorchImageEnvWrapper method) render_batch() (simulusion.envs.vector_env.VectorizedEnv method) RenderImage (class in simulusion.envs.wrappers) repeat_action_probability (simulusion.configs.iris_config.IRISConfig attribute) repeat_interleave_batch() (in module simulusion.utils.jepa_utils) ReplayBuffer (class in simulusion.datasets.diamond_dataset) (class in simulusion.memory.dreamer_memory) ResBlock (class in simulusion.models.diffusion.diamond_diffusion) reset() (simulusion.agents.planning.PlanningAgent method) (simulusion.controller.rssm_policy.RSSMPolicy method) (simulusion.envs.diamond_atari.DiamondAtariWrapper method) (simulusion.envs.dmc.DeepMindControlEnv method) (simulusion.envs.dmlab.DMLabEnv method) (simulusion.envs.gym_env.GymImageEnv method) (simulusion.envs.mujoco_env.MuJoCoImageEnv method) (simulusion.envs.procgen_env.ProcgenImageEnv method) (simulusion.envs.unity_env.UnityMLAgentsEnv method) (simulusion.envs.wrappers.FrameStack method) (simulusion.envs.wrappers.ObsDict method) (simulusion.envs.wrappers.OneHotAction method) (simulusion.envs.wrappers.RenderImage method) (simulusion.envs.wrappers.ResizeImage method) (simulusion.envs.wrappers.RewardObs method) (simulusion.envs.wrappers.TimeLimit method) (simulusion.envs.wrappers.UUID method) (simulusion.models.iris_transformer.KVCache method) (simulusion.utils.jepa_utils.AverageMeter method) (simulusion.utils.utils.TorchImageEnvWrapper method) reset_batch() (simulusion.envs.vector_env.TorchVectorizedEnv method) (simulusion.envs.vector_env.VectorizedEnv method) ResidualBlock (class in simulusion.models.diffusion.reward_termination) (class in simulusion.vision.iris_decoder) (class in simulusion.vision.iris_encoder) ResizeImage (class in simulusion.envs.wrappers) resolve_precision() (in module simulusion.inference.precision) resolve_pred_depth() (in module simulusion.helpers.jepa_helper) resolve_run_config() (in module simulusion.training.run) restart_dead_codes() (in module simulusion.vision.vq_layer) restore (simulusion.configs.dreamer_config.DreamerConfig attribute) restore_checkpoint() (simulusion.models.dreamer.Dreamer method) resume (simulusion.training.trainer.TrainerConfig attribute) reward (simulusion.inference.steppers.ImagineOutput attribute) (simulusion.world_model.protocol.Transition attribute) (simulusion.world_model.Transition attribute) reward_channels (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.diamond_config.ModelPreset attribute) reward_clip (simulusion.configs.diamond_config.DiamondConfig attribute) reward_cond_dim (simulusion.configs.diamond_config.DiamondConfig attribute) reward_loss (simulusion.configs.iris_config.IRISConfig attribute) reward_lstm_dim (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.diamond_config.ModelPreset attribute) reward_res_blocks (simulusion.configs.diamond_config.DiamondConfig attribute) reward_scale (simulusion.configs.dreamer_config.DreamerConfig attribute) reward_transform (simulusion.configs.iris_config.IRISConfig attribute) RewardObs (class in simulusion.envs.wrappers) rewards (simulusion.data.episode.Episode attribute) (simulusion.world_model.protocol.Rollout attribute) (simulusion.world_model.Rollout attribute) RewardTerminationLoss (class in simulusion.models.diffusion.reward_termination) RewardTerminationModel (class in simulusion.models.diffusion.reward_termination) rho (simulusion.configs.diamond_config.DiamondConfig attribute) RLDSKeys (class in simulusion.data.rlds) RLEnvironmentDataset (class in simulusion.datasets.video_datasets) RLLoss (class in simulusion.models.diffusion.actor_critic) Rollout (class in simulusion.world_model) (class in simulusion.world_model.protocol) rollout() (simulusion.world_model.protocol.WorldModel method) (simulusion.world_model.protocol.WorldModelBase method) (simulusion.world_model.WorldModel method) (simulusion.world_model.WorldModelBase method) rollout_drift() (in module simulusion.inference.benchmark) rollout_eval() (simulusion.controller.rollout_generator.RolloutGenerator method) rollout_eval_n() (simulusion.controller.rollout_generator.RolloutGenerator method) rollout_n() (simulusion.controller.rollout_generator.RolloutGenerator method) rollout_once() (simulusion.controller.rollout_generator.RolloutGenerator method) rollout_prior() (simulusion.models.rssm.RecurrentStateSpaceModel method) RolloutDataset (class in simulusion.datasets.wm_dataset) RolloutGenerator (class in simulusion.controller.rollout_generator) ROOT_PATH (simulusion.configs.dit_config.DiTConfig attribute) root_path (simulusion.configs.jepa_config.JEPAConfig attribute) RSSM (class in simulusion.models.dreamer_rssm) RSSMPolicy (class in simulusion.controller.rssm_policy) run() (in module simulusion.training.run) (simulusion.benchmarks.runner.BenchmarkRunner method) (simulusion.envs.vector_env.SimWorker method) run_all() (simulusion.benchmarks.runner.MultiAgentBenchmarkRunner method) run_config_dict() (in module simulusion.training.run) RUN_CONFIG_DIR (in module simulusion.training.run) run_dir (simulusion.training.trainer.Trainer attribute) (simulusion.training.trainer.TrainResult attribute) run_episode() (in module simulusion.agents.planning) run_name (simulusion.training.trainer.TrainerConfig attribute) run_training() (in module simulusion.training.run) run_training_pipeline() (in module simulusion.training.train_world_model) S sample() (simulusion.datasets.diamond_dataset.ReplayBuffer method) (simulusion.memory.dreamer_memory.Memory method) (simulusion.memory.dreamer_memory.ReplayBuffer method) (simulusion.memory.planet_memory.Memory method) (simulusion.models.diffusion.DDPM.DDPM method) (simulusion.models.diffusion.diamond_diffusion.EulerSampler method) (simulusion.models.dynamics_model.DynamicsModel method) (simulusion.models.dynamics_model.MaskGITSampler method) (simulusion.vision.dreamer_decoder.SampleDist method) sample_frame() (simulusion.models.dynamics_model.MaskGITSampler method) sample_interval (simulusion.configs.wm_config.WMVAEConfig attribute) sample_next_tokens() (simulusion.models.iris_transformer.IRISTransformer method) sample_noise_level() (simulusion.models.diffusion.diamond_diffusion.EDMPreconditioner method) sample_random_action() (simulusion.utils.utils.TorchImageEnvWrapper method) sample_sequence() (simulusion.datasets.diamond_dataset.ReplayBuffer method) (simulusion.memory.iris_memory.IRISReplayBuffer method) sample_single() (simulusion.memory.iris_memory.IRISReplayBuffer method) sample_temperature (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) sample_with_burn_in() (simulusion.memory.iris_memory.IRISReplayBuffer method) SampleDist (class in simulusion.vision.dreamer_decoder) sampling_method (simulusion.configs.diamond_config.DiamondConfig attribute) save() (simulusion.data.episode.Episode method) (simulusion.models.dreamer.Dreamer method) (simulusion.models.iris_agent.IRISAgent method) (simulusion.world_model.protocol.WorldModel method) (simulusion.world_model.protocol.WorldModelBase method) (simulusion.world_model.WorldModel method) (simulusion.world_model.WorldModelBase method) save_bundle() (in module simulusion.inference.bundle) save_checkpoint() (simulusion.training.task.TrainingTask method) (simulusion.training.train_diamond.DiamondAgent method) (simulusion.training.train_genie.GenieTrainer method) (simulusion.training.trainer.Trainer method) save_frames() (in module simulusion.utils.utils) save_interval (simulusion.configs.diamond_config.DiamondConfig attribute) save_pretrained() (simulusion.models.diffusion.DiT.DiT method) (simulusion.models.genie.Genie method) save_replay() (in module simulusion.training.tasks.common) save_video() (in module simulusion.utils.utils) scalar_freq (simulusion.configs.dreamer_config.DreamerConfig attribute) scale (simulusion.inference.quantize.Int8WeightOnlyLinear attribute) scales_loss (simulusion.training.precision.MixedPrecision property) scenario_name (simulusion.datasets.nuplan.NuPlanSample attribute) scheduler_factor (simulusion.configs.wm_config.WMMDNRNNConfig attribute) (simulusion.configs.wm_config.WMVAEConfig attribute) scheduler_patience (simulusion.configs.wm_config.WMMDNRNNConfig attribute) (simulusion.configs.wm_config.WMVAEConfig attribute) seed (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.training.tasks.common.SequenceDataConfig attribute) (simulusion.training.trainer.TrainerConfig attribute) seed() (simulusion.envs.diamond_atari.DiamondAtariWrapper method) seed_steps (simulusion.configs.dreamer_config.DreamerConfig attribute) SelectAction (class in simulusion.envs.wrappers) SelfAttentionBlock (class in simulusion.vision.iris_encoder) seq_len (simulusion.configs.wm_config.WMMDNRNNConfig attribute) seq_to_batch() (simulusion.models.dreamer_rssm.RSSM method) (simulusion.models.modular_rssm.ModularRSSM method) sequence_length (simulusion.training.tasks.common.SequenceDataConfig attribute) sequence_loaders() (in module simulusion.training.tasks.common) SequenceDataConfig (class in simulusion.training.tasks.common) SequenceDataset (class in simulusion.data.sequence) (class in simulusion.datasets.diamond_dataset) (class in simulusion.datasets.wm_dataset) set_precision() (simulusion.models.dreamer.Dreamer method) (simulusion.models.iris_agent.IRISAgent method) (simulusion.training.train_diamond.DiamondAgent method) setup() (simulusion.training.task.TrainingTask method) setup_logging() (in module simulusion.utils.logging_utils) should_stop() (simulusion.training.task.TrainingTask method) shuffle (simulusion.datasets.video_datasets.DatasetConfig attribute) sigma_data (simulusion.configs.diamond_config.DiamondConfig attribute) sigma_max (simulusion.configs.diamond_config.DiamondConfig attribute) sigma_min (simulusion.configs.diamond_config.DiamondConfig attribute) sign (simulusion.vision.dreamer_decoder.TanhBijector property) simulusion.agents.planning module simulusion.api module simulusion.benchmarks module simulusion.benchmarks.adapters module simulusion.benchmarks.metrics module simulusion.benchmarks.reporting module simulusion.benchmarks.runner module simulusion.blocks module simulusion.blocks.mhsa module simulusion.blocks.st_transformer module simulusion.catalog module simulusion.configs.diamond_config module simulusion.configs.dit_config module simulusion.configs.dreamer_config module simulusion.configs.genie_config module simulusion.configs.iris_config module simulusion.configs.jepa_config module simulusion.configs.wm_config module simulusion.controller.iris_policy module simulusion.controller.rollout_generator module simulusion.controller.rssm_policy module simulusion.data.episode module simulusion.data.folder module simulusion.data.lerobot module simulusion.data.rlds module simulusion.data.sequence module simulusion.datasets.cifar10 module simulusion.datasets.diamond_dataset module simulusion.datasets.imagenet1k module simulusion.datasets.nuplan module simulusion.datasets.tinyworlds module simulusion.datasets.video_datasets module simulusion.datasets.wm_dataset module simulusion.envs.ale_atari_env module simulusion.envs.ale_atari_vector_env module simulusion.envs.diamond_atari module simulusion.envs.dmc module simulusion.envs.dmlab module simulusion.envs.gym_env module simulusion.envs.mujoco_env module simulusion.envs.procgen_env module simulusion.envs.robotics_env module simulusion.envs.unity_env module simulusion.envs.vector_env module simulusion.envs.wrappers module simulusion.export module simulusion.helpers.jepa_helper module simulusion.inference.benchmark module simulusion.inference.bundle module simulusion.inference.precision module simulusion.inference.quantize module simulusion.inference.runtime module simulusion.inference.steppers module simulusion.losses.convae_loss module simulusion.losses.gmm_loss module simulusion.masks module simulusion.masks.default module simulusion.masks.multiblock module simulusion.masks.random module simulusion.memory.dreamer_memory module simulusion.memory.iris_memory module simulusion.memory.planet_memory module simulusion.models.controller module simulusion.models.diffusion module simulusion.models.diffusion.actor_critic module simulusion.models.diffusion.DDPM module simulusion.models.diffusion.diamond_diffusion module simulusion.models.diffusion.DiT module simulusion.models.diffusion.reward_termination module simulusion.models.dreamer module simulusion.models.dreamer_rssm module simulusion.models.dynamics_model module simulusion.models.genie module simulusion.models.iris_agent module simulusion.models.iris_transformer module simulusion.models.jepa_agent module simulusion.models.latent_action_model module simulusion.models.mdrnn module simulusion.models.modular_rssm module simulusion.models.planet module simulusion.models.rssm module simulusion.models.vit module simulusion.training module simulusion.training.precision module simulusion.training.rl_harness module simulusion.training.run module simulusion.training.task module simulusion.training.tasks.common module simulusion.training.train_controller module simulusion.training.train_convvae module simulusion.training.train_diamond module simulusion.training.train_genie module simulusion.training.train_iris module simulusion.training.train_jepa module simulusion.training.train_mdn_rnn module simulusion.training.train_planet module simulusion.training.train_rssm module simulusion.training.train_world_model module simulusion.training.trainer module simulusion.transforms.image module simulusion.utils.data_utils module simulusion.utils.dreamer_utils module simulusion.utils.jepa_utils module simulusion.utils.logging_utils module simulusion.utils.memory_utils module simulusion.utils.train_utils module simulusion.utils.utils module simulusion.vision.dreamer_decoder module simulusion.vision.dreamer_encoder module simulusion.vision.iris_decoder module simulusion.vision.iris_encoder module simulusion.vision.planet_decoder module simulusion.vision.planet_encoder module simulusion.vision.VAE.ConvVAE module simulusion.vision.video_tokenizer module simulusion.vision.vq_layer module simulusion.world_model module simulusion.world_model.diamond module simulusion.world_model.dreamer module simulusion.world_model.genie module simulusion.world_model.iris module simulusion.world_model.planet module simulusion.world_model.protocol module simulusion.world_model.vae_mdrnn module SimWorker (class in simulusion.envs.vector_env) sinusoidal_time_embedding() (in module simulusion.models.diffusion.DiT) size (simulusion.memory.planet_memory.Episode property) (simulusion.memory.planet_memory.Memory property) slice() (simulusion.data.episode.Episode method) smoothing (simulusion.agents.planning.CEMPlanner attribute) source (simulusion.data.folder.DatasetMeta attribute) (simulusion.training.tasks.common.SequenceDataConfig attribute) spec (simulusion.world_model.protocol.WorldModel attribute) (simulusion.world_model.protocol.WorldModelBase attribute) (simulusion.world_model.WorldModel attribute) (simulusion.world_model.WorldModelBase attribute) split (simulusion.datasets.tinyworlds.TinyWorldsConfig attribute) split() (simulusion.data.folder.EpisodeFolder method) start_actor_critic_after (simulusion.configs.iris_config.IRISConfig attribute) start_autoencoder_after (simulusion.configs.iris_config.IRISConfig attribute) start_lr (simulusion.configs.jepa_config.JEPAConfig attribute) start_transformer_after (simulusion.configs.iris_config.IRISConfig attribute) state (simulusion.data.rlds.RLDSKeys attribute) (simulusion.inference.steppers.ImagineOutput attribute) (simulusion.world_model.protocol.Rollout attribute) (simulusion.world_model.protocol.Transition attribute) (simulusion.world_model.Rollout attribute) (simulusion.world_model.Transition attribute) state_dict() (simulusion.datasets.diamond_dataset.ReplayBuffer method) (simulusion.inference.bundle.Bundle method) (simulusion.training.precision.GradientUpdate method) (simulusion.training.task.TrainingTask method) (simulusion.utils.train_utils.EarlyStopping method) (simulusion.utils.train_utils.ReduceLROnPlateau method) state_posterior() (simulusion.models.rssm.RecurrentStateSpaceModel method) state_prior() (simulusion.models.rssm.RecurrentStateSpaceModel method) step() (simulusion.envs.diamond_atari.DiamondAtariWrapper method) (simulusion.envs.dmc.DeepMindControlEnv method) (simulusion.envs.dmlab.DMLabEnv method) (simulusion.envs.gym_env.GymImageEnv method) (simulusion.envs.mujoco_env.MuJoCoImageEnv method) (simulusion.envs.procgen_env.ProcgenImageEnv method) (simulusion.envs.unity_env.UnityMLAgentsEnv method) (simulusion.envs.wrappers.ActionRepeat method) (simulusion.envs.wrappers.FrameStack method) (simulusion.envs.wrappers.NormalizeActions method) (simulusion.envs.wrappers.ObsDict method) (simulusion.envs.wrappers.OneHotAction method) (simulusion.envs.wrappers.RenderImage method) (simulusion.envs.wrappers.ResizeImage method) (simulusion.envs.wrappers.RewardObs method) (simulusion.envs.wrappers.SelectAction method) (simulusion.envs.wrappers.TimeLimit method) (simulusion.masks.multiblock.MaskCollator method) (simulusion.masks.random.MaskCollator method) (simulusion.utils.jepa_utils.CosineWDSchedule method) (simulusion.utils.jepa_utils.WarmupCosineSchedule method) (simulusion.utils.train_utils.EarlyStopping method) (simulusion.utils.train_utils.ReduceLROnPlateau method) (simulusion.utils.utils.TorchImageEnvWrapper method) (simulusion.world_model.protocol.WorldModel method) (simulusion.world_model.protocol.WorldModelBase method) (simulusion.world_model.WorldModel method) (simulusion.world_model.WorldModelBase method) step_batch() (simulusion.envs.vector_env.TorchVectorizedEnv method) (simulusion.envs.vector_env.VectorizedEnv method) step_module() (simulusion.inference.steppers.DreamerStepper method) steps (simulusion.inference.benchmark.DriftReport attribute) steps_per_sec (simulusion.inference.benchmark.LatencyReport attribute) STKVCache (class in simulusion.blocks.st_transformer) STMLP (class in simulusion.blocks.st_transformer) stoch_size (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.models.modular_rssm.BackboneBase attribute) (simulusion.models.modular_rssm.ModularRSSM property) stochastic (simulusion.world_model.protocol.WorldModelSpec attribute) (simulusion.world_model.WorldModelSpec attribute) stop (simulusion.utils.train_utils.EarlyStopping property) stopped_early (simulusion.training.trainer.TrainResult attribute) StreamingVideoWriter (class in simulusion.utils.utils) stride (simulusion.training.tasks.common.SequenceDataConfig attribute) STSpatialAttention (class in simulusion.blocks.st_transformer) STTemporalAttention (class in simulusion.blocks.st_transformer) STTransformer (class in simulusion.blocks.st_transformer) STTransformerBlock (class in simulusion.blocks.st_transformer) STTransformerConfig (class in simulusion.configs.genie_config) summary() (simulusion.inference.benchmark.DriftReport method) (simulusion.inference.benchmark.LatencyReport method) (simulusion.models.diffusion.DiT.DiT method) (simulusion.models.dreamer.Dreamer method) (simulusion.models.dreamer.DreamerAgent method) (simulusion.models.genie.Genie method) (simulusion.models.iris_agent.IRISAgent method) (simulusion.models.jepa_agent.JEPAAgent method) (simulusion.training.train_diamond.DiamondAgent method) supports_imagination (simulusion.inference.steppers.DreamerStepper attribute) (simulusion.inference.steppers.IRISStepper attribute) (simulusion.inference.steppers.WorldModelStepper attribute) sweep_config (simulusion.configs.jepa_config.JEPAConfig attribute) sweep_train() (in module simulusion.training.train_jepa) symexp() (in module simulusion.utils.dreamer_utils) symlog() (in module simulusion.utils.dreamer_utils) symlog_range (simulusion.configs.dreamer_config.DreamerConfig attribute) T TanhBijector (class in simulusion.vision.dreamer_decoder) target_return (simulusion.configs.wm_config.WMControllerConfig attribute) task (simulusion.training.trainer.Trainer attribute) td_lambda (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) TensorBoardMetrics (class in simulusion.utils.utils) terminate() (simulusion.memory.planet_memory.Episode method) terminate_on_life_loss (simulusion.configs.diamond_config.DiamondConfig attribute) terminated (simulusion.data.episode.Episode attribute) (simulusion.world_model.protocol.Rollout attribute) (simulusion.world_model.protocol.Transition attribute) (simulusion.world_model.Rollout attribute) (simulusion.world_model.Transition attribute) test (simulusion.configs.dreamer_config.DreamerConfig attribute) test_episodes (simulusion.configs.dreamer_config.DreamerConfig attribute) test_interval (simulusion.configs.dreamer_config.DreamerConfig attribute) test_trained_model() (in module simulusion.training.train_world_model) tf32 (simulusion.configs.dreamer_config.DreamerConfig attribute) time_limit (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.wm_config.WMControllerConfig attribute) TimeLimit (class in simulusion.envs.wrappers) TimestepEmbedder (class in simulusion.models.diffusion.DiT) TimestepEmbedding (class in simulusion.models.diffusion.diamond_diffusion) TIMESTEPS (simulusion.configs.dit_config.DiTConfig attribute) TinyWorldsConfig (class in simulusion.datasets.tinyworlds) TinyWorldsDataLoader (class in simulusion.datasets.tinyworlds) TinyWorldsDataset (class in simulusion.datasets.tinyworlds) to() (simulusion.utils.dreamer_utils.TwoHotEncoder method) to_arrays() (simulusion.data.episode.Episode method) to_channels_last() (in module simulusion.utils.memory_utils) to_dict() (simulusion.configs.jepa_config.JEPAConfig method) (simulusion.configs.wm_config.WMControllerConfig method) (simulusion.configs.wm_config.WMMDNRNNConfig method) (simulusion.configs.wm_config.WMVAEConfig method) (simulusion.world_model.protocol.WorldModelSpec method) (simulusion.world_model.WorldModelSpec method) to_float_tensor() (simulusion.training.train_iris.IRISTrainer method) to_json() (simulusion.data.folder.DatasetMeta method) to_model_domain() (in module simulusion.datasets.diamond_dataset) to_nested_dict() (simulusion.configs.jepa_config.JEPAConfig method) to_pixel_domain() (in module simulusion.datasets.diamond_dataset) to_tensor_obs() (in module simulusion.utils.utils) to_train_dict() (simulusion.configs.jepa_config.JEPAConfig method) token_drop() (simulusion.models.diffusion.DiT.LabelEmbedder method) token_embedding_dim (simulusion.configs.iris_config.IRISConfig attribute) tokenizer_decoder_depth (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) tokenizer_decoder_dim (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) tokenizer_embedding_dim (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) tokenizer_encoder_depth (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) tokenizer_encoder_dim (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) tokenizer_num_heads (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) tokenizer_vocab_size (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) tokens_per_frame (simulusion.configs.iris_config.IRISConfig attribute) TorchImageEnvWrapper (class in simulusion.utils.utils) TorchVectorizedEnv (class in simulusion.envs.vector_env) total_epochs (simulusion.configs.iris_config.IRISConfig attribute) total_steps (simulusion.configs.dreamer_config.DreamerConfig attribute) train (simulusion.configs.dreamer_config.DreamerConfig attribute) train() (in module simulusion.training.train_planet) (simulusion.models.diffusion.DiT.DiT method) (simulusion.models.dreamer.DreamerAgent method) (simulusion.models.jepa_agent.JEPAAgent method) (simulusion.models.planet.Planet method) (simulusion.training.rl_harness.PPOTrainer method) (simulusion.training.train_diamond.DiamondAgent method) (simulusion.training.train_genie.GenieTrainer method) (simulusion.training.train_iris.IRISTrainer method) train_batch_size (simulusion.configs.wm_config.WMVAEConfig attribute) train_controller() (in module simulusion.training.train_controller) train_convae() (in module simulusion.training.train_convvae) train_diamond() (in module simulusion.training.train_diamond) train_epoch() (simulusion.training.train_diamond.DiamondAgent method) (simulusion.training.train_iris.IRISTrainer method) train_mdn_rnn() (in module simulusion.training.train_mdn_rnn) train_one_batch() (simulusion.models.dreamer.Dreamer method) train_rssm() (in module simulusion.training.train_rssm) train_seq_len (simulusion.configs.dreamer_config.DreamerConfig attribute) train_step() (simulusion.training.rl_harness.PPOTrainer method) (simulusion.training.task.TrainingTask method) (simulusion.training.train_genie.GenieTrainer method) Trainer (class in simulusion.training.trainer) trainer (simulusion.training.task.TrainingTask property) TrainerConfig (class in simulusion.training.trainer) training_step() (simulusion.models.genie.Genie method) training_steps_per_epoch (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) TrainingTask (class in simulusion.training.task) TrainResult (class in simulusion.training.trainer) transform_reward() (simulusion.models.iris_agent.IRISAgent method) transformer_batch_size (simulusion.configs.iris_config.IRISConfig attribute) transformer_dropout (simulusion.configs.iris_config.IRISConfig attribute) transformer_embed_dim (simulusion.configs.iris_config.IRISConfig attribute) transformer_heads (simulusion.configs.iris_config.IRISConfig attribute) transformer_layers (simulusion.configs.iris_config.IRISConfig attribute) transformer_steps_per_epoch (simulusion.configs.iris_config.IRISConfig attribute) transformer_timesteps (simulusion.configs.iris_config.IRISConfig attribute) TransformerBackbone (class in simulusion.models.modular_rssm) TransformerBlock (class in simulusion.models.diffusion.DiT) (class in simulusion.models.modular_rssm) Transition (class in simulusion.world_model) (class in simulusion.world_model.protocol) trunc_normal_() (in module simulusion.utils.jepa_utils) truncated (simulusion.data.episode.Episode attribute) TwoHotEncoder (class in simulusion.utils.dreamer_utils) U unity_base_port (simulusion.configs.dreamer_config.DreamerConfig attribute) unity_behavior_name (simulusion.configs.dreamer_config.DreamerConfig attribute) unity_file_name (simulusion.configs.dreamer_config.DreamerConfig attribute) unity_no_graphics (simulusion.configs.dreamer_config.DreamerConfig attribute) unity_quality_level (simulusion.configs.dreamer_config.DreamerConfig attribute) unity_time_scale (simulusion.configs.dreamer_config.DreamerConfig attribute) unity_worker_id (simulusion.configs.dreamer_config.DreamerConfig attribute) UnityMLAgentsEnv (class in simulusion.envs.unity_env) unpatchify_tokens() (simulusion.models.diffusion.DiT.DiT method) UpBlock (class in simulusion.models.diffusion.diamond_diffusion) update() (simulusion.utils.jepa_utils.AverageMeter method) (simulusion.utils.utils.TensorBoardMetrics method) update_actor_critic() (simulusion.models.iris_agent.IRISAgent method) update_autoencoder() (simulusion.models.iris_agent.IRISAgent method) update_steps (simulusion.configs.dreamer_config.DreamerConfig attribute) update_transformer() (simulusion.models.iris_agent.IRISAgent method) update_world_model() (simulusion.models.dreamer.Dreamer method) UpsampleBlock (class in simulusion.vision.iris_decoder) use_amp (simulusion.configs.diamond_config.DiamondConfig attribute) (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) use_bfloat16 (simulusion.configs.jepa_config.JEPAConfig attribute) use_color_distortion (simulusion.configs.jepa_config.JEPAConfig attribute) use_cuda_graphs (simulusion.models.iris_agent.IRISAgent attribute) use_disc_model (simulusion.configs.dreamer_config.DreamerConfig attribute) use_ema (simulusion.configs.genie_config.VideoTokenizerConfig attribute) use_gaussian_blur (simulusion.configs.jepa_config.JEPAConfig attribute) use_horizontal_flip (simulusion.configs.jepa_config.JEPAConfig attribute) UUID (class in simulusion.envs.wrappers) V val_batches (simulusion.training.tasks.common.SequenceDataConfig attribute) val_fraction (simulusion.training.tasks.common.SequenceDataConfig attribute) VAL_SPLIT (simulusion.configs.dit_config.DiTConfig attribute) val_split (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.configs.jepa_config.JEPAConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) validate() (simulusion.configs.wm_config.WMControllerConfig method) (simulusion.configs.wm_config.WMMDNRNNConfig method) (simulusion.configs.wm_config.WMVAEConfig method) (simulusion.data.episode.Episode method) (simulusion.training.train_genie.GenieTrainer method) validate_epoch() (simulusion.training.train_genie.GenieTrainer method) value_learning_rate (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) value_loss() (simulusion.models.diffusion.actor_critic.RLLoss method) (simulusion.models.dreamer.Dreamer method) VectorizedEnv (class in simulusion.envs.vector_env) VectorQuantizer (class in simulusion.vision.vq_layer) VectorQuantizerEMA (class in simulusion.vision.vq_layer) verify_export() (in module simulusion.export) video_format (simulusion.configs.dreamer_config.DreamerConfig attribute) video_fps (simulusion.configs.dreamer_config.DreamerConfig attribute) video_paths (simulusion.datasets.video_datasets.VideoDatasetBase attribute) VideoDataset (class in simulusion.training.train_genie) VideoDatasetBase (class in simulusion.datasets.video_datasets) VideoDatasetConfig (class in simulusion.datasets.video_datasets) VideoFolderDataset (class in simulusion.datasets.video_datasets) VideoTokenizer (class in simulusion.vision.video_tokenizer) VideoTokenizerConfig (class in simulusion.configs.genie_config) VisionTransformer (class in simulusion.models.vit) VisionTransformerPredictor (class in simulusion.models.vit) visualize_latent_tsne() (in module simulusion.utils.utils) visualize_latent_umap() (in module simulusion.utils.utils) vit_base() (in module simulusion.models.vit) vit_giant() (in module simulusion.models.vit) vit_huge() (in module simulusion.models.vit) vit_large() (in module simulusion.models.vit) vit_predictor() (in module simulusion.models.vit) vit_small() (in module simulusion.models.vit) vit_tiny() (in module simulusion.models.vit) ViTEncoder (class in simulusion.models.modular_rssm) vocab_size (simulusion.configs.genie_config.DynamicsModelConfig attribute) (simulusion.configs.genie_config.LatentActionModelConfig attribute) (simulusion.configs.genie_config.VideoTokenizerConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) W wandb_entity (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.jepa_config.JEPAConfig attribute) wandb_project (simulusion.configs.dreamer_config.DreamerConfig attribute) (simulusion.configs.jepa_config.JEPAConfig attribute) warmup (simulusion.configs.jepa_config.JEPAConfig attribute) warmup() (simulusion.models.planet.Planet method) warmup_steps (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) WarmupCosineSchedule (class in simulusion.utils.jepa_utils) WEIGHT_DECAY (simulusion.configs.dit_config.DiTConfig attribute) weight_decay (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.configs.iris_config.IRISConfig attribute) (simulusion.configs.jepa_config.JEPAConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) weight_decay_actor (simulusion.configs.diamond_config.DiamondConfig attribute) weight_decay_diffusion (simulusion.configs.diamond_config.DiamondConfig attribute) weight_decay_reward (simulusion.configs.diamond_config.DiamondConfig attribute) weight_int8 (simulusion.inference.quantize.Int8WeightOnlyLinear attribute) weight_memory_bytes() (in module simulusion.inference.quantize) WIDTH (simulusion.configs.dit_config.DiTConfig attribute) width (simulusion.configs.wm_config.WMVAEConfig attribute) window_attention_heads (simulusion.configs.genie_config.GenieConfig attribute) (simulusion.configs.genie_config.GenieSmallConfig attribute) (simulusion.configs.genie_config.LatentActionModelConfig attribute) (simulusion.training.train_genie.GenieConfig attribute) within() (simulusion.inference.benchmark.DriftReport method) WMControllerConfig (class in simulusion.configs.wm_config) WMMDNRNNConfig (class in simulusion.configs.wm_config) WMVAEConfig (class in simulusion.configs.wm_config) WORKDIR (simulusion.configs.dit_config.DiTConfig attribute) world_model() (simulusion.training.task.TrainingTask method) world_model_class() (in module simulusion.world_model) world_model_loss() (simulusion.models.dreamer.Dreamer method) world_model_path (simulusion.training.trainer.TrainResult attribute) WorldModel (class in simulusion.world_model) (class in simulusion.world_model.protocol) WorldModelBase (class in simulusion.world_model) (class in simulusion.world_model.protocol) WorldModelSpec (class in simulusion.world_model) (class in simulusion.world_model.protocol) WorldModelStepper (class in simulusion.inference.steppers) worst_rel_error (simulusion.inference.benchmark.DriftReport property) write() (simulusion.data.folder.EpisodeWriter method) (simulusion.utils.utils.Mp4Writer method) write_frame() (simulusion.utils.utils.StreamingVideoWriter method) write_meta() (in module simulusion.data.folder) write_tag (simulusion.configs.jepa_config.JEPAConfig attribute)
