Index

A

act() (synora.controller.iris_policy.IRISPolicy method) (synora.inference.steppers.DreamerStepper method) (synora.inference.steppers.IRISStepper method) (synora.inference.steppers.WorldModelStepper method) (synora.models.iris_agent.IRISAgent method) act_and_collect_data() (synora.models.dreamer.Dreamer method) act_with_world_model() (synora.models.dreamer.Dreamer method) action_decoder_dim (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) action_embedding_dim (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) action_encoder_depth (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) action_encoder_dim (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) action_noise (synora.configs.dreamer_config.DreamerConfig attribute) action_num_heads (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) action_pooling (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.configs.genie_config.LatentActionModelConfig attribute) (synora.training.train_genie.GenieConfig attribute) action_repeat (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.iris_config.IRISConfig attribute) action_size (synora.configs.wm_config.WMControllerConfig attribute) (synora.configs.wm_config.WMMDNRNNConfig attribute) (synora.utils.utils.TorchImageEnvWrapper property) action_space (synora.envs.dmc.DeepMindControlEnv property) (synora.envs.dmlab.DMLabEnv property) (synora.envs.gym_env.GymImageEnv property) (synora.envs.mujoco_env.MuJoCoImageEnv property) (synora.envs.procgen_env.ProcgenImageEnv property) (synora.envs.unity_env.UnityMLAgentsEnv property) (synora.envs.wrappers.FrameStack property) (synora.envs.wrappers.NormalizeActions property) (synora.envs.wrappers.ObsDict property) (synora.envs.wrappers.OneHotAction property) action_vocab_size (synora.configs.genie_config.DynamicsModelConfig attribute) (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) ActionDecoder (class in synora.vision.dreamer_decoder) ActionRepeat (class in synora.envs.wrappers) actor_channels (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.diamond_config.ModelPreset attribute) actor_critic_batch_size (synora.configs.iris_config.IRISConfig attribute) actor_critic_steps_per_epoch (synora.configs.iris_config.IRISConfig attribute) actor_hidden_size (synora.configs.iris_config.IRISConfig attribute) actor_layers (synora.configs.iris_config.IRISConfig attribute) actor_learning_rate (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.iris_config.IRISConfig attribute)

actor_loss() (synora.models.dreamer.Dreamer method) actor_lstm_dim (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.diamond_config.ModelPreset attribute) actor_res_blocks (synora.configs.diamond_config.DiamondConfig attribute) ActorCritic (class in synora.training.rl_harness) ActorCriticNetwork (class in synora.models.diffusion.actor_critic) adam_beta1 (synora.configs.iris_config.IRISConfig attribute) adam_beta2 (synora.configs.iris_config.IRISConfig attribute) adam_epsilon (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.dreamer_config.DreamerConfig attribute) AdaptiveGroupNorm (class in synora.models.diffusion.diamond_diffusion) add() (synora.datasets.diamond_dataset.ReplayBuffer method) (synora.memory.dreamer_memory.ReplayBuffer method) (synora.memory.iris_memory.IRISOnPolicyBuffer method) (synora.memory.iris_memory.IRISReplayBuffer method) add_exploration() (synora.vision.dreamer_decoder.ActionDecoder method) advance() (synora.blocks.st_transformer.STKVCache method) (synora.models.iris_transformer.KVCache method) agent_types (synora.datasets.nuplan.NuPlanSample attribute) agents_future (synora.datasets.nuplan.NuPlanSample attribute) agents_mask (synora.datasets.nuplan.NuPlanSample attribute) agents_past (synora.datasets.nuplan.NuPlanSample attribute) algo (synora.configs.dreamer_config.DreamerConfig attribute) aliases (synora.api.EnvBackendSpec attribute) (synora.api.ModelSpec attribute) AllGather (class in synora.utils.jepa_utils) allow_overlap (synora.configs.jepa_config.JEPAConfig attribute) AllReduce (class in synora.utils.jepa_utils) AllReduceSum (class in synora.utils.jepa_utils) append() (synora.blocks.st_transformer.STKVCache method) (synora.memory.dreamer_memory.Episode method) (synora.memory.dreamer_memory.Memory method) (synora.memory.planet_memory.Episode method) (synora.memory.planet_memory.Memory method) (synora.models.iris_transformer.KVCache method) apply_gradient_checkpointing() (in module synora.utils.memory_utils) apply_masks() (in module synora.utils.utils) apply_model() (in module synora.utils.utils) artifacts (synora.inference.bundle.Bundle attribute) as_dict() (synora.inference.benchmark.DriftReport method) (synora.inference.benchmark.LatencyReport method) aspect_ratio (synora.configs.jepa_config.JEPAConfig attribute) assert_finite() (in module synora.utils.logging_utils) assert_finite_values() (in module synora.utils.logging_utils) assign_type() (synora.utils.utils.TensorBoardMetrics method) atanh() (synora.vision.dreamer_decoder.TanhBijector method) atari_100k (synora.configs.iris_config.IRISConfig attribute) Attention (class in synora.models.vit) AttentionBlock (class in synora.models.diffusion.diamond_diffusion) attn_drop_rate (synora.configs.genie_config.DynamicsModelConfig attribute) (synora.configs.genie_config.STTransformerConfig attribute) AttrDict (class in synora.utils.utils) autoencoder_batch_size (synora.configs.iris_config.IRISConfig attribute) autoregressive_sample() (synora.models.dynamics_model.DynamicsModel method) AverageMeter (class in synora.utils.jepa_utils)

B

BackboneBase (class in synora.models.modular_rssm) backward() (synora.utils.jepa_utils.AllGather static method) (synora.utils.jepa_utils.AllReduce static method) (synora.utils.jepa_utils.AllReduceSum static method) BaseAdapter (class in synora.benchmarks.adapters) BATCH (synora.configs.dit_config.DiTConfig attribute) batch_size (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.configs.jepa_config.JEPAConfig attribute) (synora.configs.wm_config.WMMDNRNNConfig attribute) (synora.datasets.tinyworlds.TinyWorldsConfig attribute) (synora.datasets.video_datasets.DatasetConfig attribute) (synora.inference.benchmark.LatencyReport attribute) (synora.training.train_genie.GenieConfig attribute) benchmark_step() (in module synora.inference.benchmark) BenchmarkRunner (class in synora.benchmarks.runner)

BETA_END (synora.configs.dit_config.DiTConfig attribute) BETA_START (synora.configs.dit_config.DiTConfig attribute) Block (class in synora.models.vit) bootstrap_ci() (in module synora.benchmarks.metrics) bootstrap_iqm_ci() (in module synora.benchmarks.metrics) bottle() (in module synora.utils.utils) brax_auto_reset (synora.configs.dreamer_config.DreamerConfig attribute) brax_backend (synora.configs.dreamer_config.DreamerConfig attribute) brax_jit (synora.configs.dreamer_config.DreamerConfig attribute) brax_suppress_warp_warnings (synora.configs.dreamer_config.DreamerConfig attribute) buffer_capacity (synora.memory.iris_memory.IRISReplayBuffer property) buffer_size (synora.configs.dreamer_config.DreamerConfig attribute) build_loss_fn() (in module synora.training.train_jepa) build_parser() (in module synora.training.train_planet) Bundle (class in synora.inference.bundle) burn_in() (synora.models.iris_agent.IRISAgent method) burn_in_length (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.iris_config.IRISConfig attribute)

C

cache_dir (synora.datasets.tinyworlds.TinyWorldsConfig attribute) canonical_dit_key() (in module synora.configs.dit_config) CausalSelfAttention (class in synora.models.iris_transformer) CHANNELS (synora.configs.dit_config.DiTConfig attribute) checkpoint_dir (synora.configs.diamond_config.DiamondConfig attribute) CHECKPOINT_EVERY (synora.configs.dit_config.DiTConfig attribute) CHECKPOINT_FORMAT (synora.models.genie.Genie attribute) (synora.models.iris_agent.IRISAgent attribute) checkpoint_interval (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.iris_config.IRISConfig attribute) checkpoint_path (synora.configs.dreamer_config.DreamerConfig attribute) CLASS_DROPOUT_PROB (synora.configs.dit_config.DiTConfig attribute) classes (synora.datasets.imagenet1k.ImageNetSubset property) clear() (synora.memory.iris_memory.IRISOnPolicyBuffer method) close() (synora.envs.diamond_atari.DiamondAtariWrapper method) (synora.envs.dmlab.DMLabEnv method) (synora.envs.gym_env.GymImageEnv method) (synora.envs.mujoco_env.MuJoCoImageEnv method) (synora.envs.procgen_env.ProcgenImageEnv method) (synora.envs.unity_env.UnityMLAgentsEnv method) (synora.envs.vector_env.VectorizedEnv method) (synora.utils.logging_utils.MetricsLogger method) (synora.utils.utils.StreamingVideoWriter method) (synora.utils.utils.TorchImageEnvWrapper method) cnn_activation_function (synora.configs.dreamer_config.DreamerConfig attribute) CNNDecoder (class in synora.vision.planet_decoder) CNNEncoder (class in synora.vision.planet_encoder) CNNFeatureExtractor (class in synora.controller.iris_policy) collect_epsilon (synora.configs.iris_config.IRISConfig attribute) collect_experience() (synora.training.train_iris.IRISTrainer method) collect_random_episodes() (synora.models.dreamer.Dreamer method) collect_steps (synora.configs.dreamer_config.DreamerConfig attribute) collect_system_stats() (in module synora.utils.logging_utils) collect_temperature (synora.configs.iris_config.IRISConfig attribute) collect_trajectories() (synora.training.rl_harness.PPOTrainer method) collection_epochs (synora.configs.iris_config.IRISConfig attribute) color_jitter_strength (synora.configs.jepa_config.JEPAConfig attribute) combine_videos() (in module synora.utils.utils) commitment_weight (synora.configs.genie_config.LatentActionModelConfig attribute) (synora.configs.genie_config.VideoTokenizerConfig attribute) (synora.configs.iris_config.IRISConfig attribute) compute_aggregate_metrics() (in module synora.benchmarks.metrics)

compute_gae() (synora.training.rl_harness.PPOTrainer method) compute_lambda_return() (in module synora.models.iris_agent) compute_lambda_returns() (synora.models.diffusion.actor_critic.RLLoss method) compute_return() (in module synora.utils.dreamer_utils) config (synora.inference.bundle.Bundle attribute) config_path (synora.api.ModelSpec attribute) continue_prob (synora.inference.steppers.ImagineOutput attribute) Controller (class in synora.models.controller) conv_vae_loss_fn() (in module synora.losses.convae_loss) ConvDecoder (class in synora.models.modular_rssm) (class in synora.vision.dreamer_decoder) ConvEmbed (class in synora.models.vit) ConvEncoder (class in synora.models.modular_rssm) (class in synora.vision.dreamer_encoder) ConvVAE (class in synora.vision.VAE.ConvVAE) ConvVAEDecoder (class in synora.vision.VAE.ConvVAE) ConvVAEEncoder (class in synora.vision.VAE.ConvVAE) copy_data (synora.configs.jepa_config.JEPAConfig attribute) copy_imgnt_locally() (in module synora.datasets.imagenet1k) CosineWDSchedule (class in synora.utils.jepa_utils) create_config() (in module synora.api) create_dataloader() (synora.datasets.tinyworlds.TinyWorldsDataLoader static method) create_dit() (in module synora.models.diffusion.DiT) create_dynamics_model() (in module synora.models.dynamics_model) create_efficient_dataloader() (in module synora.utils.data_utils) create_genie() (in module synora.models.genie) create_genie_large() (in module synora.models.genie) create_genie_small() (in module synora.models.genie) create_genie_trainer() (in module synora.training.train_genie) create_latent_action_model() (in module synora.models.latent_action_model) create_model() (in module synora.api) create_modular_rssm() (in module synora.models.modular_rssm) create_rl_harness_example() (in module synora.training.rl_harness) create_st_transformer() (in module synora.blocks.st_transformer) create_tinyworlds_dataloader() (in module synora.datasets.tinyworlds) create_video_dataloader() (in module synora.datasets.video_datasets) create_video_dataset_from_config() (in module synora.datasets.video_datasets) create_video_tokenizer() (in module synora.vision.video_tokenizer) crop_scale (synora.configs.jepa_config.JEPAConfig attribute) CROP_SIZE (synora.configs.dit_config.DiTConfig attribute) crop_size (synora.configs.jepa_config.JEPAConfig attribute) CSVLogger (class in synora.utils.jepa_utils)

D

data_dir (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.wm_config.WMMDNRNNConfig attribute) (synora.configs.wm_config.WMVAEConfig attribute) data_loader_num_workers (synora.configs.diamond_config.DiamondConfig attribute) data_pass() (in module synora.training.train_mdn_rnn) data_source (synora.datasets.video_datasets.VideoDatasetBase attribute) (synora.datasets.video_datasets.VideoDatasetConfig attribute) DATASET (synora.configs.dit_config.DiTConfig attribute) dataset (synora.configs.jepa_config.JEPAConfig attribute) DATASET_CONFIGS (synora.datasets.tinyworlds.TinyWorldsDataset attribute) dataset_name (synora.datasets.tinyworlds.TinyWorldsConfig attribute) DATASET_NAMES (synora.datasets.tinyworlds.TinyWorldsDataLoader attribute) dataset_type (synora.datasets.video_datasets.VideoDatasetConfig attribute) DatasetConfig (class in synora.datasets.video_datasets) DDPM (class in synora.models.diffusion.DDPM) decode() (synora.inference.steppers.DreamerStepper method) (synora.models.latent_action_model.LatentActionModel method) (synora.utils.dreamer_utils.TwoHotEncoder method) (synora.vision.iris_decoder.DiscreteAutoencoder method) (synora.vision.video_tokenizer.VideoTokenizer method) decode_from_embeddings() (synora.vision.iris_decoder.IRISDecoder method) decode_from_indices() (synora.vision.iris_decoder.IRISDecoder method) (synora.vision.iris_encoder.IRISEncoder method) decode_indices() (synora.vision.video_tokenizer.VideoTokenizer method) (synora.vision.vq_layer.VectorQuantizer method) (synora.vision.vq_layer.VectorQuantizerEMA method) decode_observation() (synora.models.modular_rssm.ModularRSSM method) decode_reward() (synora.models.modular_rssm.ModularRSSM method) decode_tokens() (synora.models.iris_transformer.IRISWorldModel method) decoder_depth (synora.configs.genie_config.VideoTokenizerConfig attribute) (synora.configs.iris_config.IRISConfig attribute) decoder_dim (synora.configs.genie_config.VideoTokenizerConfig attribute) DecoderBase (class in synora.models.modular_rssm) DeepMindControlEnv (class in synora.envs.dmc) default_collect_temperature() (in module synora.training.train_iris) DefaultCollator (class in synora.masks) (class in synora.masks.default) denoise() (synora.models.diffusion.diamond_diffusion.EDMPreconditioner method) dense_activation_function (synora.configs.dreamer_config.DreamerConfig attribute) DenseDecoder (class in synora.vision.dreamer_decoder) DEPTH (synora.configs.dit_config.DiTConfig attribute) depth (synora.configs.genie_config.DynamicsModelConfig attribute) (synora.configs.genie_config.STTransformerConfig attribute) dequantized_weight() (synora.inference.quantize.Int8WeightOnlyLinear method) description (synora.api.EnvBackendSpec attribute) (synora.api.ModelSpec attribute) detach_state() (synora.models.dreamer_rssm.RSSM method) (synora.models.modular_rssm.ModularRSSM method) detect_anomaly (synora.configs.dreamer_config.DreamerConfig attribute) deter_size (synora.configs.dreamer_config.DreamerConfig attribute) (synora.models.modular_rssm.BackboneBase attribute) (synora.models.modular_rssm.ModularRSSM property) deterministic_state_fwd() (synora.models.rssm.RecurrentStateSpaceModel method) device (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.wm_config.WMMDNRNNConfig attribute) (synora.configs.wm_config.WMVAEConfig attribute) (synora.inference.runtime.InferenceModel property) (synora.inference.steppers.WorldModelStepper attribute) DiamondAdapter (class in synora.benchmarks.adapters)

DiamondAgent (class in synora.training.train_diamond) DiamondAtariWrapper (class in synora.envs.diamond_atari) DiamondConfig (class in synora.configs.diamond_config) diffusion_channels (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.diamond_config.ModelPreset attribute) diffusion_cond_dim (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.diamond_config.ModelPreset attribute) diffusion_res_blocks (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.diamond_config.ModelPreset attribute) DiffusionUNet (class in synora.models.diffusion.diamond_diffusion) dim (synora.configs.genie_config.DynamicsModelConfig attribute) (synora.configs.genie_config.STTransformerConfig attribute) disc_loss_coeff (synora.configs.dreamer_config.DreamerConfig attribute) discount (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.iris_config.IRISConfig attribute) discount_factor (synora.configs.diamond_config.DiamondConfig attribute) DiscreteAutoencoder (class in synora.vision.iris_decoder) display (synora.configs.wm_config.WMControllerConfig attribute) DiT (class in synora.models.diffusion.DiT) dit_preset_config() (in module synora.configs.dit_config) DiTConfig (class in synora.configs.dit_config) dmlab_action_repeat (synora.configs.dreamer_config.DreamerConfig attribute) dmlab_action_set (synora.configs.dreamer_config.DreamerConfig attribute) dmlab_config (synora.configs.dreamer_config.DreamerConfig attribute) dmlab_observations (synora.configs.dreamer_config.DreamerConfig attribute) dmlab_renderer (synora.configs.dreamer_config.DreamerConfig attribute) DMLabEnv (class in synora.envs.dmlab) DownBlock (class in synora.models.diffusion.diamond_diffusion) download (synora.configs.jepa_config.JEPAConfig attribute) download_all_datasets() (in module synora.datasets.tinyworlds) Dreamer (class in synora.models.dreamer) DreamerAdapter (class in synora.benchmarks.adapters) DreamerAgent (class in synora.models.dreamer) DreamerConfig (class in synora.configs.dreamer_config) DreamerPolicyExport (class in synora.export) DreamerStepModule (class in synora.inference.steppers) DreamerStepper (class in synora.inference.steppers) DreamerV1Adapter (class in synora.benchmarks.adapters) DreamerV2Adapter (class in synora.benchmarks.adapters) DriftReport (class in synora.inference.benchmark) DROP (synora.configs.dit_config.DiTConfig attribute) drop_path() (in module synora.models.vit) drop_path_rate (synora.configs.genie_config.DynamicsModelConfig attribute) (synora.configs.genie_config.STTransformerConfig attribute) drop_rate (synora.configs.genie_config.DynamicsModelConfig attribute) (synora.configs.genie_config.STTransformerConfig attribute) DropPath (class in synora.blocks.st_transformer) (class in synora.models.vit) dump_scalars_to_pickle() (synora.utils.dreamer_utils.Logger method) dynamics_depth (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) dynamics_dim (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) dynamics_num_heads (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) DynamicsModel (class in synora.models.dynamics_model) DynamicsModelConfig (class in synora.configs.genie_config)

E

early_stopping (synora.configs.diamond_config.DiamondConfig attribute) EARLY_STOPPING (synora.configs.dit_config.DiTConfig attribute) early_stopping (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.configs.iris_config.IRISConfig attribute) (synora.configs.jepa_config.JEPAConfig attribute) (synora.training.train_genie.GenieConfig attribute) early_stopping_patience (synora.configs.wm_config.WMMDNRNNConfig attribute) (synora.configs.wm_config.WMVAEConfig attribute) EarlyStopping (class in synora.utils.train_utils) EDMPreconditioner (class in synora.models.diffusion.diamond_diffusion) ego_future (synora.datasets.nuplan.NuPlanSample attribute) ego_past (synora.datasets.nuplan.NuPlanSample attribute) EMA (synora.configs.dit_config.DiTConfig attribute) ema (synora.configs.jepa_config.JEPAConfig attribute) EMA_DECAY (synora.configs.dit_config.DiTConfig attribute) ema_decay (synora.configs.genie_config.VideoTokenizerConfig attribute) embed_size (synora.models.modular_rssm.EncoderBase attribute) (synora.models.modular_rssm.ModularRSSM property) embedding_dim (synora.configs.genie_config.DynamicsModelConfig attribute) (synora.configs.genie_config.LatentActionModelConfig attribute) (synora.configs.genie_config.VideoTokenizerConfig attribute) embedding_size (synora.models.modular_rssm.GRUBackbone property) (synora.models.modular_rssm.LSTMBackbone property) (synora.models.modular_rssm.TransformerBackbone property) enable_console_metrics (synora.configs.dreamer_config.DreamerConfig attribute) enable_jsonl (synora.configs.dreamer_config.DreamerConfig attribute) enable_mixed_precision() (in module synora.utils.memory_utils) enable_performance_defaults() (in module synora.utils.memory_utils) enable_sweep (synora.configs.jepa_config.JEPAConfig attribute) enable_tensorboard (synora.configs.dreamer_config.DreamerConfig attribute) enable_wandb (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.jepa_config.JEPAConfig attribute) enc_mask_scale (synora.configs.jepa_config.JEPAConfig attribute) encode() (synora.models.latent_action_model.LatentActionModel method) (synora.utils.dreamer_utils.TwoHotEncoder method) (synora.vision.iris_decoder.DiscreteAutoencoder method) (synora.vision.video_tokenizer.VideoTokenizer method) encode_to_indices() (synora.vision.iris_encoder.IRISEncoder method) encode_video() (synora.models.genie.Genie method) encoder_channels (synora.configs.iris_config.IRISConfig attribute) encoder_depth (synora.configs.genie_config.LatentActionModelConfig attribute) (synora.configs.genie_config.VideoTokenizerConfig attribute) encoder_dim (synora.configs.genie_config.LatentActionModelConfig attribute) (synora.configs.genie_config.VideoTokenizerConfig attribute) encoder_layers (synora.configs.iris_config.IRISConfig attribute) encoder_residual_blocks (synora.configs.iris_config.IRISConfig attribute)

EncoderBase (class in synora.models.modular_rssm) ensure_results_dir_exists() (in module synora.utils.utils) entropy() (synora.vision.dreamer_decoder.SampleDist method) entropy_coef (synora.configs.iris_config.IRISConfig attribute) entropy_weight (synora.configs.diamond_config.DiamondConfig attribute) env (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.iris_config.IRISConfig attribute) env_backend (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.iris_config.IRISConfig attribute) env_instance (synora.configs.dreamer_config.DreamerConfig attribute) env_name (synora.configs.wm_config.WMControllerConfig attribute) env_steps_per_epoch (synora.configs.iris_config.IRISConfig attribute) EnvBackendSpec (class in synora.api) environment_steps_per_epoch (synora.configs.diamond_config.DiamondConfig attribute) Episode (class in synora.memory.dreamer_memory) (class in synora.memory.planet_memory) episode_boundaries() (synora.datasets.diamond_dataset.ReplayBuffer method) EPOCHS (synora.configs.dit_config.DiTConfig attribute) epochs (synora.configs.jepa_config.JEPAConfig attribute) epsilon_greedy (synora.configs.diamond_config.DiamondConfig attribute) EulerSampler (class in synora.models.diffusion.diamond_diffusion) eval_episodes (synora.configs.iris_config.IRISConfig attribute) eval_interval (synora.configs.diamond_config.DiamondConfig attribute) eval_temperature (synora.configs.iris_config.IRISConfig attribute) evaluate (synora.configs.dreamer_config.DreamerConfig attribute) evaluate() (in module synora.training.train_controller) (in module synora.training.train_rssm) (synora.benchmarks.adapters.BaseAdapter method) (synora.benchmarks.adapters.DiamondAdapter method) (synora.benchmarks.adapters.DreamerAdapter method) (synora.benchmarks.adapters.IRISAdapter method) (synora.models.dreamer.Dreamer method) (synora.models.dreamer.DreamerAgent method) (synora.training.train_diamond.DiamondAgent method) (synora.training.train_iris.IRISTrainer method) example_inputs() (synora.inference.steppers.DreamerStepModule method) exp_name (synora.configs.dreamer_config.DreamerConfig attribute) expected_reward() (synora.models.iris_transformer.IRISTransformer method) experience_replay (synora.configs.dreamer_config.DreamerConfig attribute) export() (synora.export.ExportableAgentMixin method) export_any() (in module synora.export) export_csv() (in module synora.benchmarks.reporting) export_latex() (in module synora.benchmarks.reporting) export_markdown() (in module synora.benchmarks.reporting) export_model() (in module synora.export) ExportableAgentMixin (class in synora.export) extensions (synora.datasets.video_datasets.VideoDatasetConfig attribute) extra_repr() (synora.inference.quantize.Int8WeightOnlyLinear method)

F

factory_path (synora.api.EnvBackendSpec attribute) features() (synora.inference.steppers.DreamerStepper static method) filter_dataset_() (synora.datasets.imagenet1k.ImageNetSubset method) final_lr (synora.configs.jepa_config.JEPAConfig attribute) final_max_abs_error (synora.inference.benchmark.DriftReport property) final_weight_decay (synora.configs.jepa_config.JEPAConfig attribute) FinalLayer (class in synora.models.diffusion.DiT) fit() (synora.models.diffusion.DiT.DiT class method) fix_init_weight() (synora.models.vit.VisionTransformer method) (synora.models.vit.VisionTransformerPredictor method) flatten_dict() (in module synora.utils.utils) flatten_parameters() (in module synora.training.train_controller) flush() (synora.utils.dreamer_utils.Logger method) (synora.utils.logging_utils.MetricsLogger method) folder (synora.configs.jepa_config.JEPAConfig attribute) formats (synora.inference.bundle.Bundle property) forward() (synora.blocks.mhsa.MultiHeadSelfAttention method) (synora.blocks.st_transformer.DropPath method) (synora.blocks.st_transformer.STMLP method) (synora.blocks.st_transformer.STSpatialAttention method) (synora.blocks.st_transformer.STTemporalAttention method) (synora.blocks.st_transformer.STTransformer method) (synora.blocks.st_transformer.STTransformerBlock method) (synora.controller.iris_policy.CNNFeatureExtractor method) (synora.controller.iris_policy.IRISActor method) (synora.controller.iris_policy.IRISCritic method) (synora.controller.iris_policy.IRISPolicy method) (synora.export.DreamerPolicyExport method) (synora.export.IRISActorCriticExport method) (synora.inference.quantize.Int8WeightOnlyLinear method) (synora.inference.runtime.InferenceModel method) (synora.inference.steppers.DreamerStepModule method) (synora.models.controller.Controller method) (synora.models.diffusion.actor_critic.ActorCriticNetwork method) (synora.models.diffusion.diamond_diffusion.AdaptiveGroupNorm method) (synora.models.diffusion.diamond_diffusion.AttentionBlock method) (synora.models.diffusion.diamond_diffusion.DiffusionUNet method) (synora.models.diffusion.diamond_diffusion.DownBlock method) (synora.models.diffusion.diamond_diffusion.ResBlock method) (synora.models.diffusion.diamond_diffusion.TimestepEmbedding method) (synora.models.diffusion.diamond_diffusion.UpBlock method) (synora.models.diffusion.DiT.DiT method) (synora.models.diffusion.DiT.FinalLayer method) (synora.models.diffusion.DiT.LabelEmbedder method) (synora.models.diffusion.DiT.PatchEmbed method) (synora.models.diffusion.DiT.PatchUnEmbed method) (synora.models.diffusion.DiT.TimestepEmbedder method) (synora.models.diffusion.DiT.TransformerBlock method) (synora.models.diffusion.reward_termination.ResidualBlock method) (synora.models.diffusion.reward_termination.RewardTerminationLoss method) (synora.models.diffusion.reward_termination.RewardTerminationModel method) (synora.models.dreamer_rssm.RSSM method) (synora.models.dynamics_model.DynamicsModel method) (synora.models.genie.Genie method) (synora.models.iris_transformer.CausalSelfAttention method) (synora.models.iris_transformer.GPTBlock method) (synora.models.iris_transformer.IRISTransformer method) (synora.models.iris_transformer.IRISWorldModel method) (synora.models.latent_action_model.LatentActionModel method) (synora.models.mdrnn.MDRNN method) (synora.models.mdrnn.MDRNNCell method) (synora.models.modular_rssm.BackboneBase method) (synora.models.modular_rssm.ConvDecoder method) (synora.models.modular_rssm.ConvEncoder method) (synora.models.modular_rssm.DecoderBase method) (synora.models.modular_rssm.EncoderBase method) (synora.models.modular_rssm.GRUBackbone method) (synora.models.modular_rssm.LSTMBackbone method) (synora.models.modular_rssm.MLPDecoder method) (synora.models.modular_rssm.MLPEncoder method) (synora.models.modular_rssm.TransformerBackbone method) (synora.models.modular_rssm.TransformerBlock method) (synora.models.modular_rssm.ViTEncoder method) (synora.models.rssm.RecurrentStateSpaceModel method) (synora.models.vit.Attention method) (synora.models.vit.Block method) (synora.models.vit.ConvEmbed method) (synora.models.vit.DropPath method) (synora.models.vit.MLP method) (synora.models.vit.PatchEmbed method) (synora.models.vit.VisionTransformer method) (synora.models.vit.VisionTransformerPredictor method) (synora.training.rl_harness.ActorCritic method) (synora.utils.jepa_utils.AllGather static method) (synora.utils.jepa_utils.AllReduce static method) (synora.utils.jepa_utils.AllReduceSum static method) (synora.vision.dreamer_decoder.ActionDecoder method) (synora.vision.dreamer_decoder.ConvDecoder method) (synora.vision.dreamer_decoder.DenseDecoder method) (synora.vision.dreamer_encoder.ConvEncoder method) (synora.vision.iris_decoder.DiscreteAutoencoder method) (synora.vision.iris_decoder.IRISDecoder method) (synora.vision.iris_decoder.ResidualBlock method) (synora.vision.iris_decoder.UpsampleBlock method) (synora.vision.iris_encoder.IRISEncoder method) (synora.vision.iris_encoder.ResidualBlock method) (synora.vision.iris_encoder.SelfAttentionBlock method) (synora.vision.planet_decoder.CNNDecoder method) (synora.vision.planet_encoder.CNNEncoder method) (synora.vision.VAE.ConvVAE.ConvVAE method) (synora.vision.VAE.ConvVAE.ConvVAEDecoder method) (synora.vision.VAE.ConvVAE.ConvVAEEncoder method) (synora.vision.video_tokenizer.VideoTokenizer method) (synora.vision.vq_layer.VectorQuantizer method) (synora.vision.vq_layer.VectorQuantizerEMA method)

forward_actor_critic() (synora.models.iris_agent.IRISAgent method) forward_cached() (synora.models.dynamics_model.DynamicsModel method) forward_with_cfg() (synora.models.diffusion.DiT.DiT method) frame_channels (synora.configs.iris_config.IRISConfig attribute) frame_height (synora.configs.iris_config.IRISConfig attribute) frame_stack (synora.configs.dreamer_config.DreamerConfig attribute) frame_width (synora.configs.iris_config.IRISConfig attribute) frameskip (synora.configs.diamond_config.DiamondConfig attribute) FrameStack (class in synora.envs.wrappers) free_nats (synora.configs.dreamer_config.DreamerConfig attribute) FreezeParameters (class in synora.utils.dreamer_utils) from_config() (synora.models.diffusion.DiT.DiT class method) (synora.models.dreamer.Dreamer class method) (synora.models.dreamer.DreamerAgent class method) (synora.models.genie.Genie class method) (synora.models.iris_agent.IRISAgent class method) (synora.models.jepa_agent.JEPAAgent class method) (synora.training.train_diamond.DiamondAgent class method) from_dict() (synora.configs.jepa_config.JEPAConfig class method) from_linear() (synora.inference.quantize.Int8WeightOnlyLinear class method) from_pretrained() (synora.models.diffusion.DiT.DiT class method) (synora.models.dreamer.Dreamer class method) (synora.models.dreamer.DreamerAgent class method) (synora.models.genie.Genie class method) (synora.models.iris_agent.IRISAgent class method) (synora.models.jepa_agent.JEPAAgent class method) (synora.training.train_diamond.DiamondAgent class method)

G

game (synora.configs.diamond_config.DiamondConfig attribute) GaussianBlur (class in synora.transforms.image) generate() (synora.models.genie.Genie method) generate_frame_cached() (synora.models.iris_transformer.IRISTransformer method) generate_rollouts() (in module synora.training.train_world_model) Genie (class in synora.models.genie) genie_kwargs_from_config() (in module synora.models.genie) GenieConfig (class in synora.configs.genie_config) (class in synora.training.train_genie) GenieSmallConfig (class in synora.configs.genie_config) GenieTrainer (class in synora.training.train_genie) get_1d_sincos_pos_embed() (in module synora.models.vit) get_1d_sincos_pos_embed_from_grid() (in module synora.models.vit) get_2d_sincos_pos_embed() (in module synora.models.diffusion.DiT) (in module synora.models.vit) get_2d_sincos_pos_embed_from_grid() (in module synora.models.vit) get_action() (synora.controller.iris_policy.IRISActor method) (synora.models.diffusion.actor_critic.ActorCriticNetwork method) (synora.training.rl_harness.ActorCritic method) get_actions() (synora.models.diffusion.actor_critic.ActorCriticNetwork method) get_arrays() (synora.memory.iris_memory.IRISOnPolicyBuffer method) get_autoencoder_config() (synora.configs.iris_config.IRISConfig method) get_available_memory() (in module synora.models.dreamer) get_combined_params() (in module synora.utils.utils) get_dataset_info() (synora.datasets.tinyworlds.TinyWorldsDataLoader static method) get_default_device() (in module synora.configs.diamond_config) get_dist() (synora.models.dreamer_rssm.RSSM method) (synora.models.modular_rssm.ModularRSSM method) get_dit_config() (in module synora.configs.dit_config) get_embed_size() (synora.models.modular_rssm.EncoderBase method) get_env_backend_spec() (in module synora.api) get_epsilon() (synora.training.train_iris.IRISTrainer method)

get_frame_shape() (synora.configs.iris_config.IRISConfig method) get_hidden_size() (synora.models.diffusion.actor_critic.ActorCriticNetwork method) get_info() (synora.datasets.tinyworlds.TinyWorldsDataset method) get_init_hidden() (synora.models.mdrnn.MDRNN method) (synora.models.mdrnn.MDRNNCell method) get_init_state() (synora.models.rssm.RecurrentStateSpaceModel method) get_intermediate_layers() (synora.models.vit.VisionTransformer method) get_loss() (in module synora.training.train_mdn_rnn) get_mask() (in module synora.utils.utils) get_mask_prob() (synora.models.dynamics_model.MaskGITSampler method) get_model_spec() (in module synora.api) get_num_parameters() (synora.models.genie.Genie method) get_package_logger() (in module synora.utils.logging_utils) get_parameters() (in module synora.utils.dreamer_utils) get_posterior() (synora.models.dreamer_rssm.RSSM method) get_preconditioners() (synora.models.diffusion.diamond_diffusion.EDMPreconditioner method) get_prior() (synora.models.dreamer_rssm.RSSM method) get_rl_config() (synora.configs.iris_config.IRISConfig method) get_transformer_config() (synora.configs.iris_config.IRISConfig method) get_value() (synora.models.diffusion.actor_critic.ActorCriticNetwork method) gmm_components (synora.configs.wm_config.WMMDNRNNConfig attribute) gmm_loss() (in module synora.losses.gmm_loss) gpt_blocks() (synora.models.iris_transformer.IRISTransformer method) GPTBlock (class in synora.models.iris_transformer) gpu_timer() (in module synora.utils.jepa_utils) grad_clip_norm (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.iris_config.IRISConfig attribute) grad_logger() (in module synora.utils.jepa_utils) gradient_checkpointing (synora.configs.iris_config.IRISConfig attribute) GRUBackbone (class in synora.models.modular_rssm) gym_render_mode (synora.configs.dreamer_config.DreamerConfig attribute) GymImageEnv (class in synora.envs.gym_env)

H

HDF5Dataset (class in synora.datasets.video_datasets) HEADS (synora.configs.dit_config.DiTConfig attribute)

height (synora.configs.wm_config.WMVAEConfig attribute) hidden_size (synora.configs.wm_config.WMControllerConfig attribute) (synora.configs.wm_config.WMMDNRNNConfig attribute)

I

image_folder (synora.configs.jepa_config.JEPAConfig attribute) image_size (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.genie_config.DynamicsModelConfig attribute) (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.configs.genie_config.LatentActionModelConfig attribute) (synora.configs.genie_config.VideoTokenizerConfig attribute) (synora.datasets.tinyworlds.TinyWorldsConfig attribute) (synora.datasets.video_datasets.DatasetConfig attribute) (synora.training.train_genie.GenieConfig attribute) ImageFolderDataset (class in synora.datasets.video_datasets) ImageNet (class in synora.datasets.imagenet1k) ImageNetSubset (class in synora.datasets.imagenet1k) imagination_horizon (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.iris_config.IRISConfig attribute) imagine() (synora.inference.steppers.DreamerStepper method) (synora.models.iris_transformer.IRISWorldModel method) imagine_horizon (synora.configs.dreamer_config.DreamerConfig attribute) imagine_rollout() (synora.models.dreamer_rssm.RSSM method) (synora.models.iris_agent.IRISAgent method) (synora.models.modular_rssm.ModularRSSM method) imagine_step() (synora.models.dreamer_rssm.RSSM method) (synora.models.iris_transformer.IRISTransformer method) (synora.models.modular_rssm.ModularRSSM method) ImagineOutput (class in synora.inference.steppers) IMG_SIZE (synora.configs.dit_config.DiTConfig attribute) import_path (synora.api.ModelSpec attribute) in_channels (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.configs.genie_config.LatentActionModelConfig attribute) (synora.configs.genie_config.VideoTokenizerConfig attribute) (synora.training.train_genie.GenieConfig attribute) include_reward (synora.configs.wm_config.WMMDNRNNConfig attribute) infer_actions() (synora.models.genie.Genie method) inference_context() (in module synora.inference.precision) InferenceModel (class in synora.inference.runtime) init_cache() (synora.blocks.st_transformer.STTransformer method) (synora.models.dynamics_model.DynamicsModel method) (synora.models.iris_transformer.IRISTransformer method)

init_distributed() (in module synora.utils.jepa_utils) init_hidden() (synora.controller.iris_policy.IRISPolicy method) (synora.models.diffusion.actor_critic.ActorCriticNetwork method) (synora.models.diffusion.reward_termination.RewardTerminationModel method) init_hidden_state() (synora.controller.iris_policy.IRISActor method) (synora.controller.iris_policy.IRISCritic method) init_model() (in module synora.helpers.jepa_helper) init_opt() (in module synora.helpers.jepa_helper) init_state() (synora.inference.steppers.DreamerStepper method) (synora.inference.steppers.IRISStepper method) (synora.inference.steppers.WorldModelStepper method) (synora.models.dreamer_rssm.RSSM method) (synora.models.modular_rssm.BackboneBase method) (synora.models.modular_rssm.GRUBackbone method) (synora.models.modular_rssm.LSTMBackbone method) (synora.models.modular_rssm.ModularRSSM method) (synora.models.modular_rssm.TransformerBackbone method) Int8WeightOnlyLinear (class in synora.inference.quantize) interpolate_pos_encoding() (synora.models.vit.VisionTransformer method) ipe_scale (synora.configs.jepa_config.JEPAConfig attribute) iqm_of_array() (in module synora.benchmarks.metrics) IRISActor (class in synora.controller.iris_policy) IRISActorCriticExport (class in synora.export) IRISAdapter (class in synora.benchmarks.adapters) IRISAgent (class in synora.models.iris_agent) IRISConfig (class in synora.configs.iris_config) IRISCritic (class in synora.controller.iris_policy) IRISDecoder (class in synora.vision.iris_decoder) IRISEncoder (class in synora.vision.iris_encoder) IRISOnPolicyBuffer (class in synora.memory.iris_memory) IRISPolicy (class in synora.controller.iris_policy) IRISReplayBuffer (class in synora.memory.iris_memory) IRISStepper (class in synora.inference.steppers) IRISTrainer (class in synora.training.train_iris) IRISTransformer (class in synora.models.iris_transformer) IRISWorldModel (class in synora.models.iris_transformer) is_moved_mujoco_error() (in module synora.envs.robotics_env) is_ready() (synora.datasets.diamond_dataset.ReplayBuffer method) items_per_sec (synora.inference.benchmark.LatencyReport attribute) iterations (synora.inference.benchmark.LatencyReport attribute)

J

JEPAAgent (class in synora.models.jepa_agent)

JEPAConfig (class in synora.configs.jepa_config) jsonl_filename (synora.configs.dreamer_config.DreamerConfig attribute)

K

kl_alpha (synora.configs.dreamer_config.DreamerConfig attribute)

kl_loss_coeff (synora.configs.dreamer_config.DreamerConfig attribute) KVCache (class in synora.models.iris_transformer)

L

LabelEmbedder (class in synora.models.diffusion.DiT) lambda_returns (synora.configs.diamond_config.DiamondConfig attribute) LatencyReport (class in synora.inference.benchmark) latent_size (synora.configs.wm_config.WMControllerConfig attribute) (synora.configs.wm_config.WMMDNRNNConfig attribute) (synora.configs.wm_config.WMVAEConfig attribute) LatentActionModel (class in synora.models.latent_action_model) LatentActionModelConfig (class in synora.configs.genie_config) LatentSequenceDataset (class in synora.datasets.wm_dataset) LEARN_SIGMA (synora.configs.dit_config.DiTConfig attribute) learning_rate (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.configs.wm_config.WMMDNRNNConfig attribute) (synora.configs.wm_config.WMVAEConfig attribute) (synora.training.train_genie.GenieConfig attribute) lineplot() (in module synora.utils.utils) list_available_atari_envs() (in module synora.envs.ale_atari_env) list_available_datasets() (synora.datasets.tinyworlds.TinyWorldsDataLoader static method) list_dit_presets() (in module synora.configs.dit_config) list_env_backends() (in module synora.api) list_envs() (in module synora.api) list_gymnasium_robotics_envs() (in module synora.envs.robotics_env) list_models() (in module synora.api) list_procgen_envs() (in module synora.envs.procgen_env) load() (synora.models.iris_agent.IRISAgent method) load_artifact() (synora.inference.bundle.Bundle method) load_bundle() (in module synora.inference.bundle) load_checkpoint (synora.configs.jepa_config.JEPAConfig attribute) load_checkpoint() (in module synora.helpers.jepa_helper) (synora.benchmarks.adapters.BaseAdapter method) (synora.benchmarks.adapters.DiamondAdapter method) (synora.benchmarks.adapters.DreamerAdapter method) (synora.benchmarks.adapters.IRISAdapter method) (synora.training.train_diamond.DiamondAgent method) (synora.training.train_genie.GenieTrainer method)

load_exported() (in module synora.export) load_memory() (in module synora.utils.utils) load_next_buffer() (synora.datasets.wm_dataset.RolloutDataset method) load_parameters() (in module synora.training.train_controller) load_state_dict() (synora.datasets.diamond_dataset.ReplayBuffer method) (synora.utils.train_utils.EarlyStopping method) (synora.utils.train_utils.ReduceLROnPlateau method) load_weights() (synora.inference.bundle.Bundle method) load_yml_config() (in module synora.utils.utils) log() (synora.utils.jepa_utils.CSVLogger method) (synora.utils.logging_utils.MetricsLogger method) log_abs_det_jacobian() (synora.vision.dreamer_decoder.TanhBijector method) log_dir (synora.configs.dreamer_config.DreamerConfig attribute) log_file (synora.configs.dreamer_config.DreamerConfig attribute) log_interval (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.iris_config.IRISConfig attribute) log_level (synora.configs.dreamer_config.DreamerConfig attribute) log_scalar() (synora.utils.dreamer_utils.Logger method) log_scalars() (synora.utils.dreamer_utils.Logger method) log_system_stats_freq (synora.configs.dreamer_config.DreamerConfig attribute) log_video() (synora.utils.logging_utils.MetricsLogger method) log_video_freq (synora.configs.dreamer_config.DreamerConfig attribute) log_videos() (synora.utils.dreamer_utils.Logger method) logdir (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.wm_config.WMControllerConfig attribute) (synora.configs.wm_config.WMMDNRNNConfig attribute) (synora.configs.wm_config.WMVAEConfig attribute) Logger (class in synora.utils.dreamer_utils) loss_type (synora.configs.jepa_config.JEPAConfig attribute) LR (synora.configs.dit_config.DiTConfig attribute) lr (synora.configs.jepa_config.JEPAConfig attribute) (synora.utils.train_utils.ReduceLROnPlateau property) lr_reference_batch_size (synora.configs.jepa_config.JEPAConfig attribute) LSTMBackbone (class in synora.models.modular_rssm)

M

main() (in module synora.training.train_diamond) (in module synora.training.train_genie) (in module synora.training.train_iris) (in module synora.training.train_jepa) (in module synora.training.train_planet) (in module synora.training.train_rssm) (in module synora.training.train_world_model) main_from_cli() (in module synora.training.train_jepa) make_atari_env() (in module synora.envs.ale_atari_env) make_atari_vector_env() (in module synora.envs.ale_atari_vector_env) make_cifar10() (in module synora.datasets.cifar10) make_diamond_atari_env() (in module synora.envs.diamond_atari) make_dmc_env() (in module synora.envs.dmc) make_dmlab_env() (in module synora.envs.dmlab) make_env() (in module synora.api) (in module synora.models.dreamer) make_gym_env() (in module synora.envs.gym_env) make_gymnasium_env_with_robotics_fallback() (in module synora.envs.robotics_env) make_imagefolder() (in module synora.datasets.imagenet1k) make_imagenet1k() (in module synora.datasets.imagenet1k) make_mujoco_env() (in module synora.envs.mujoco_env) make_mujoco_env_from_config() (in module synora.envs.mujoco_env) make_nuplan_dataloader() (in module synora.datasets.nuplan) make_procgen_env() (in module synora.envs.procgen_env) make_robotics_env() (in module synora.envs.robotics_env) make_stepper() (in module synora.inference.steppers) make_transforms() (in module synora.transforms.image) make_unity_mlagents_env() (in module synora.envs.unity_env) manifest (synora.inference.bundle.Bundle attribute) map_raster (synora.datasets.nuplan.NuPlanSample attribute) mask_prob_max (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) mask_prob_min (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) MaskCollator (class in synora.masks.multiblock) (class in synora.masks.random) maskgit_steps (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) MaskGITSampler (class in synora.models.dynamics_model) max_abs_error (synora.inference.benchmark.DriftReport attribute) max_env_steps (synora.configs.iris_config.IRISConfig attribute) max_episode_length (synora.configs.dreamer_config.DreamerConfig attribute) max_episode_steps (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.iris_config.IRISConfig attribute) (synora.envs.dmlab.DMLabEnv property) (synora.envs.gym_env.GymImageEnv property) (synora.envs.procgen_env.ProcgenImageEnv property) (synora.envs.unity_env.UnityMLAgentsEnv property) (synora.utils.utils.TorchImageEnvWrapper property) max_noop (synora.configs.diamond_config.DiamondConfig attribute) max_steps (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) max_videos_to_save (synora.configs.dreamer_config.DreamerConfig attribute) max_workers (synora.configs.wm_config.WMControllerConfig attribute) maybe_compile() (in module synora.utils.memory_utils) MDRNN (class in synora.models.mdrnn) MDRNNCell (class in synora.models.mdrnn) mean() (synora.vision.dreamer_decoder.SampleDist method) mean_action() (synora.vision.dreamer_decoder.ActionDecoder method) mean_ms (synora.inference.benchmark.LatencyReport attribute) Memory (class in synora.memory.dreamer_memory) (class in synora.memory.planet_memory) MetricsLogger (class in synora.utils.logging_utils) min_delta (synora.configs.diamond_config.DiamondConfig attribute) MIN_DELTA (synora.configs.dit_config.DiTConfig attribute) min_delta (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.configs.iris_config.IRISConfig attribute) (synora.configs.jepa_config.JEPAConfig attribute) (synora.training.train_genie.GenieConfig attribute) min_keep (synora.configs.jepa_config.JEPAConfig attribute) min_ms (synora.inference.benchmark.LatencyReport attribute) MLP (class in synora.models.vit) mlp_ratio (synora.configs.genie_config.DynamicsModelConfig attribute) (synora.configs.genie_config.STTransformerConfig attribute) MLPDecoder (class in synora.models.modular_rssm) MLPEncoder (class in synora.models.modular_rssm) mode() (synora.vision.dreamer_decoder.SampleDist method) model_learning_rate (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.iris_config.IRISConfig attribute) model_name (synora.configs.jepa_config.JEPAConfig attribute) ModelPreset (class in synora.configs.diamond_config) ModelSpec (class in synora.api) ModularRSSM (class in synora.models.modular_rssm) modulate() (in module synora.models.diffusion.DiT) module synora.api synora.benchmarks synora.benchmarks.adapters synora.benchmarks.metrics synora.benchmarks.reporting synora.benchmarks.runner synora.blocks synora.blocks.mhsa synora.blocks.st_transformer synora.catalog synora.configs.diamond_config synora.configs.dit_config synora.configs.dreamer_config synora.configs.genie_config synora.configs.iris_config synora.configs.jepa_config synora.configs.wm_config synora.controller.iris_policy synora.controller.rollout_generator synora.controller.rssm_policy synora.datasets.cifar10 synora.datasets.diamond_dataset synora.datasets.imagenet1k synora.datasets.nuplan synora.datasets.tinyworlds synora.datasets.video_datasets synora.datasets.wm_dataset synora.envs.ale_atari_env synora.envs.ale_atari_vector_env synora.envs.diamond_atari synora.envs.dmc synora.envs.dmlab synora.envs.gym_env synora.envs.mujoco_env synora.envs.procgen_env synora.envs.robotics_env synora.envs.unity_env synora.envs.vector_env synora.envs.wrappers synora.export synora.helpers.jepa_helper synora.inference.benchmark synora.inference.bundle synora.inference.precision synora.inference.quantize synora.inference.runtime synora.inference.steppers synora.losses.convae_loss synora.losses.gmm_loss synora.masks synora.masks.default synora.masks.multiblock synora.masks.random synora.memory.dreamer_memory synora.memory.iris_memory synora.memory.planet_memory synora.models.controller synora.models.diffusion synora.models.diffusion.actor_critic synora.models.diffusion.DDPM synora.models.diffusion.diamond_diffusion synora.models.diffusion.DiT synora.models.diffusion.reward_termination synora.models.dreamer synora.models.dreamer_rssm synora.models.dynamics_model synora.models.genie synora.models.iris_agent synora.models.iris_transformer synora.models.jepa_agent synora.models.latent_action_model synora.models.mdrnn synora.models.modular_rssm synora.models.planet synora.models.rssm synora.models.vit synora.training synora.training.rl_harness synora.training.train_controller synora.training.train_convvae synora.training.train_diamond synora.training.train_genie synora.training.train_iris synora.training.train_jepa synora.training.train_mdn_rnn synora.training.train_planet synora.training.train_rssm synora.training.train_world_model synora.transforms.image synora.utils.data_utils synora.utils.dreamer_utils synora.utils.jepa_utils synora.utils.logging_utils synora.utils.memory_utils synora.utils.train_utils synora.utils.utils synora.vision.dreamer_decoder synora.vision.dreamer_encoder synora.vision.iris_decoder synora.vision.iris_encoder synora.vision.planet_decoder synora.vision.planet_encoder synora.vision.VAE.ConvVAE synora.vision.video_tokenizer synora.vision.vq_layer

Mp4Writer (class in synora.utils.utils) mujoco_binary_path (synora.configs.dreamer_config.DreamerConfig attribute) mujoco_camera (synora.configs.dreamer_config.DreamerConfig attribute) mujoco_frame_skip (synora.configs.dreamer_config.DreamerConfig attribute) mujoco_reset_noise_scale (synora.configs.dreamer_config.DreamerConfig attribute) mujoco_xml_path (synora.configs.dreamer_config.DreamerConfig attribute) mujoco_xml_string (synora.configs.dreamer_config.DreamerConfig attribute) MuJoCoImageEnv (class in synora.envs.mujoco_env) MultiAgentBenchmarkRunner (class in synora.benchmarks.runner) MultiblockMaskCollator (in module synora.masks) MultiHeadSelfAttention (class in synora.blocks.mhsa)

N

n_samples (synora.configs.wm_config.WMControllerConfig attribute) name (synora.api.EnvBackendSpec attribute) (synora.api.ModelSpec attribute) (synora.inference.benchmark.LatencyReport attribute) (synora.vision.dreamer_decoder.SampleDist property) no_gpu (synora.configs.dreamer_config.DreamerConfig attribute) noreload (synora.configs.wm_config.WMMDNRNNConfig attribute) (synora.configs.wm_config.WMVAEConfig attribute) normalize_frames_for_saving() (in module synora.utils.utils) normalize_procgen_env_name() (in module synora.envs.procgen_env) NormalizeActions (class in synora.envs.wrappers) nosamples (synora.configs.wm_config.WMVAEConfig attribute) num_buckets (synora.configs.dreamer_config.DreamerConfig attribute) NUM_CLASSES (synora.configs.dit_config.DiTConfig attribute) num_conditioning_frames (synora.configs.diamond_config.DiamondConfig attribute) num_enc_masks (synora.configs.jepa_config.JEPAConfig attribute) num_epochs (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.wm_config.WMMDNRNNConfig attribute) (synora.configs.wm_config.WMVAEConfig attribute) num_frames (synora.configs.genie_config.DynamicsModelConfig attribute) (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.configs.genie_config.LatentActionModelConfig attribute) (synora.configs.genie_config.STTransformerConfig attribute) (synora.configs.genie_config.VideoTokenizerConfig attribute) (synora.datasets.tinyworlds.TinyWorldsConfig attribute) (synora.datasets.video_datasets.DatasetConfig attribute) (synora.training.train_genie.GenieConfig attribute)

num_heads (synora.configs.genie_config.DynamicsModelConfig attribute) (synora.configs.genie_config.LatentActionModelConfig attribute) (synora.configs.genie_config.STTransformerConfig attribute) (synora.configs.genie_config.VideoTokenizerConfig attribute) num_patches_per_frame (synora.configs.genie_config.STTransformerConfig attribute) num_pred_masks (synora.configs.jepa_config.JEPAConfig attribute) num_sampling_steps (synora.configs.diamond_config.DiamondConfig attribute) num_units (synora.configs.dreamer_config.DreamerConfig attribute) NUM_WORKERS (synora.configs.dit_config.DiTConfig attribute) num_workers (synora.configs.jepa_config.JEPAConfig attribute) (synora.datasets.tinyworlds.TinyWorldsConfig attribute) (synora.datasets.video_datasets.DatasetConfig attribute) NumPyDataset (class in synora.datasets.video_datasets) NuPlanDataset (class in synora.datasets.nuplan) NuPlanSample (class in synora.datasets.nuplan)

O

obs_embed_size (synora.configs.dreamer_config.DreamerConfig attribute) obs_key (synora.datasets.video_datasets.VideoDatasetConfig attribute) obs_size (synora.configs.diamond_config.DiamondConfig attribute) obs_space (synora.envs.wrappers.RenderImage property) (synora.envs.wrappers.ResizeImage property) ObsDict (class in synora.envs.wrappers) observation_size (synora.utils.utils.TorchImageEnvWrapper property) observation_space (synora.envs.dmc.DeepMindControlEnv property) (synora.envs.dmlab.DMLabEnv property) (synora.envs.gym_env.GymImageEnv property) (synora.envs.mujoco_env.MuJoCoImageEnv property) (synora.envs.procgen_env.ProcgenImageEnv property) (synora.envs.unity_env.UnityMLAgentsEnv property) (synora.envs.wrappers.FrameStack property) (synora.envs.wrappers.ObsDict property) (synora.envs.wrappers.RewardObs property)

ObservationDataset (class in synora.datasets.wm_dataset) observe() (synora.inference.steppers.DreamerStepper method) (synora.inference.steppers.IRISStepper method) (synora.inference.steppers.WorldModelStepper method) observe_rollout() (synora.models.dreamer_rssm.RSSM method) (synora.models.modular_rssm.ModularRSSM method) observe_step() (synora.models.dreamer_rssm.RSSM method) (synora.models.modular_rssm.ModularRSSM method) OneHotAction (class in synora.envs.wrappers) optimize_for_inference() (in module synora.inference.runtime) optimize_memory_efficient_ops() (in module synora.utils.memory_utils)

P

p50_ms (synora.inference.benchmark.LatencyReport attribute) p90_ms (synora.inference.benchmark.LatencyReport attribute) p99_ms (synora.inference.benchmark.LatencyReport attribute) p_mean (synora.configs.diamond_config.DiamondConfig attribute) p_sample() (synora.models.diffusion.DDPM.DDPM method) p_std (synora.configs.diamond_config.DiamondConfig attribute) parameter_count() (synora.models.diffusion.DiT.DiT method) (synora.models.dreamer.Dreamer method) (synora.models.dreamer.DreamerAgent method) (synora.models.genie.Genie method) (synora.models.iris_agent.IRISAgent method) (synora.models.jepa_agent.JEPAAgent method) (synora.training.train_diamond.DiamondAgent method) PATCH (synora.configs.dit_config.DiTConfig attribute) patch_size (synora.configs.genie_config.DynamicsModelConfig attribute) (synora.configs.genie_config.LatentActionModelConfig attribute) (synora.configs.genie_config.VideoTokenizerConfig attribute) (synora.configs.jepa_config.JEPAConfig attribute) PatchEmbed (class in synora.models.diffusion.DiT) (class in synora.models.vit) PatchUnEmbed (class in synora.models.diffusion.DiT) path (synora.inference.bundle.Bundle attribute) patience (synora.configs.diamond_config.DiamondConfig attribute) PATIENCE (synora.configs.dit_config.DiTConfig attribute) patience (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.configs.iris_config.IRISConfig attribute) (synora.configs.jepa_config.JEPAConfig attribute) (synora.training.train_genie.GenieConfig attribute) peak_memory_mb (synora.inference.benchmark.LatencyReport attribute) perceptual_blocks (synora.configs.iris_config.IRISConfig attribute) perceptual_linear_weights (synora.configs.iris_config.IRISConfig attribute)

perceptual_weight (synora.configs.iris_config.IRISConfig attribute) perf_defaults (synora.configs.dreamer_config.DreamerConfig attribute) persistent_workers (synora.configs.diamond_config.DiamondConfig attribute) pin_mem (synora.configs.jepa_config.JEPAConfig attribute) pin_memory (synora.configs.diamond_config.DiamondConfig attribute) (synora.datasets.video_datasets.DatasetConfig attribute) Planet (class in synora.models.planet) planning_target (synora.datasets.nuplan.NuPlanSample attribute) play() (synora.models.genie.Genie method) plot_metrics() (in module synora.utils.utils) policy_loss() (synora.models.diffusion.actor_critic.RLLoss method) poll() (synora.controller.rssm_policy.RSSMPolicy method) pop_size (synora.configs.wm_config.WMControllerConfig attribute) postprocess_img() (in module synora.utils.utils) PPOTrainer (class in synora.training.rl_harness) precision_dtype() (in module synora.inference.precision) precompute_latents() (in module synora.training.train_mdn_rnn) pred_depth (synora.configs.jepa_config.JEPAConfig attribute) pred_emb_dim (synora.configs.jepa_config.JEPAConfig attribute) pred_mask_scale (synora.configs.jepa_config.JEPAConfig attribute) pred_reward() (synora.models.rssm.RecurrentStateSpaceModel method) predict() (synora.models.diffusion.reward_termination.RewardTerminationModel method) predict_next_tokens() (synora.models.iris_transformer.IRISTransformer method) prefetch_iterator() (in module synora.utils.data_utils) preprocess_frame() (synora.training.train_iris.IRISTrainer method) preprocess_img() (in module synora.utils.utils) preprocess_obs() (in module synora.models.dreamer) preset (synora.configs.diamond_config.DiamondConfig attribute) prime_cache() (synora.models.iris_transformer.IRISTransformer method) procgen_distribution_mode (synora.configs.dreamer_config.DreamerConfig attribute) procgen_num_levels (synora.configs.dreamer_config.DreamerConfig attribute) procgen_start_level (synora.configs.dreamer_config.DreamerConfig attribute) ProcgenImageEnv (class in synora.envs.procgen_env)

Q

q_sample() (synora.models.diffusion.DDPM.DDPM method) qkv_bias (synora.configs.genie_config.DynamicsModelConfig attribute) (synora.configs.genie_config.STTransformerConfig attribute)

quantize_weights() (in module synora.inference.quantize) quantizer (synora.configs.iris_config.IRISConfig attribute)

R

RandomMaskCollator (in module synora.masks) read_checkpoint (synora.configs.jepa_config.JEPAConfig attribute) reconstruct() (synora.models.iris_agent.IRISAgent method) reconstruction_weight (synora.configs.iris_config.IRISConfig attribute) RecurrentStateSpaceModel (class in synora.models.rssm) recursive (synora.datasets.video_datasets.VideoDatasetConfig attribute) ReduceLROnPlateau (class in synora.utils.train_utils) register_buffers() (synora.utils.dreamer_utils.TwoHotEncoder method) register_gymnasium_robotics_envs() (in module synora.envs.robotics_env) rel_error (synora.inference.benchmark.DriftReport attribute) release() (synora.utils.utils.Mp4Writer method) render (synora.configs.dreamer_config.DreamerConfig attribute) render() (synora.envs.diamond_atari.DiamondAtariWrapper method) (synora.envs.dmc.DeepMindControlEnv method) (synora.envs.dmlab.DMLabEnv method) (synora.envs.gym_env.GymImageEnv method) (synora.envs.mujoco_env.MuJoCoImageEnv method) (synora.envs.procgen_env.ProcgenImageEnv method) (synora.envs.unity_env.UnityMLAgentsEnv method) (synora.utils.utils.TorchImageEnvWrapper method) render_batch() (synora.envs.vector_env.VectorizedEnv method) RenderImage (class in synora.envs.wrappers) repeat_action_probability (synora.configs.iris_config.IRISConfig attribute) repeat_interleave_batch() (in module synora.utils.jepa_utils) ReplayBuffer (class in synora.datasets.diamond_dataset) (class in synora.memory.dreamer_memory) ResBlock (class in synora.models.diffusion.diamond_diffusion) reset() (synora.controller.rssm_policy.RSSMPolicy method) (synora.envs.diamond_atari.DiamondAtariWrapper method) (synora.envs.dmc.DeepMindControlEnv method) (synora.envs.dmlab.DMLabEnv method) (synora.envs.gym_env.GymImageEnv method) (synora.envs.mujoco_env.MuJoCoImageEnv method) (synora.envs.procgen_env.ProcgenImageEnv method) (synora.envs.unity_env.UnityMLAgentsEnv method) (synora.envs.wrappers.FrameStack method) (synora.envs.wrappers.ObsDict method) (synora.envs.wrappers.OneHotAction method) (synora.envs.wrappers.RenderImage method) (synora.envs.wrappers.ResizeImage method) (synora.envs.wrappers.RewardObs method) (synora.envs.wrappers.TimeLimit method) (synora.envs.wrappers.UUID method) (synora.utils.jepa_utils.AverageMeter method) (synora.utils.utils.TorchImageEnvWrapper method)

reset_batch() (synora.envs.vector_env.TorchVectorizedEnv method) (synora.envs.vector_env.VectorizedEnv method) ResidualBlock (class in synora.models.diffusion.reward_termination) (class in synora.vision.iris_decoder) (class in synora.vision.iris_encoder) ResizeImage (class in synora.envs.wrappers) resolve_precision() (in module synora.inference.precision) resolve_pred_depth() (in module synora.helpers.jepa_helper) restart_dead_codes() (in module synora.vision.vq_layer) restore (synora.configs.dreamer_config.DreamerConfig attribute) restore_checkpoint() (synora.models.dreamer.Dreamer method) reward (synora.inference.steppers.ImagineOutput attribute) reward_channels (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.diamond_config.ModelPreset attribute) reward_clip (synora.configs.diamond_config.DiamondConfig attribute) reward_cond_dim (synora.configs.diamond_config.DiamondConfig attribute) reward_loss (synora.configs.iris_config.IRISConfig attribute) reward_lstm_dim (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.diamond_config.ModelPreset attribute) reward_res_blocks (synora.configs.diamond_config.DiamondConfig attribute) reward_scale (synora.configs.dreamer_config.DreamerConfig attribute) reward_transform (synora.configs.iris_config.IRISConfig attribute) RewardObs (class in synora.envs.wrappers) RewardTerminationLoss (class in synora.models.diffusion.reward_termination) RewardTerminationModel (class in synora.models.diffusion.reward_termination) rho (synora.configs.diamond_config.DiamondConfig attribute) RLEnvironmentDataset (class in synora.datasets.video_datasets) RLLoss (class in synora.models.diffusion.actor_critic) rollout_drift() (in module synora.inference.benchmark) rollout_eval() (synora.controller.rollout_generator.RolloutGenerator method) rollout_eval_n() (synora.controller.rollout_generator.RolloutGenerator method) rollout_n() (synora.controller.rollout_generator.RolloutGenerator method) rollout_once() (synora.controller.rollout_generator.RolloutGenerator method) rollout_prior() (synora.models.rssm.RecurrentStateSpaceModel method) RolloutDataset (class in synora.datasets.wm_dataset) RolloutGenerator (class in synora.controller.rollout_generator) ROOT_PATH (synora.configs.dit_config.DiTConfig attribute) root_path (synora.configs.jepa_config.JEPAConfig attribute) RSSM (class in synora.models.dreamer_rssm) RSSMPolicy (class in synora.controller.rssm_policy) run() (synora.benchmarks.runner.BenchmarkRunner method) (synora.envs.vector_env.SimWorker method) run_all() (synora.benchmarks.runner.MultiAgentBenchmarkRunner method) run_training_pipeline() (in module synora.training.train_world_model)

S

sample() (synora.datasets.diamond_dataset.ReplayBuffer method) (synora.memory.dreamer_memory.Memory method) (synora.memory.dreamer_memory.ReplayBuffer method) (synora.memory.planet_memory.Memory method) (synora.models.diffusion.DDPM.DDPM method) (synora.models.diffusion.diamond_diffusion.EulerSampler method) (synora.models.dynamics_model.DynamicsModel method) (synora.models.dynamics_model.MaskGITSampler method) (synora.vision.dreamer_decoder.SampleDist method) sample_frame() (synora.models.dynamics_model.MaskGITSampler method) sample_interval (synora.configs.wm_config.WMVAEConfig attribute) sample_next_tokens() (synora.models.iris_transformer.IRISTransformer method) sample_noise_level() (synora.models.diffusion.diamond_diffusion.EDMPreconditioner method) sample_random_action() (synora.utils.utils.TorchImageEnvWrapper method) sample_sequence() (synora.datasets.diamond_dataset.ReplayBuffer method) (synora.memory.iris_memory.IRISReplayBuffer method) sample_single() (synora.memory.iris_memory.IRISReplayBuffer method) sample_temperature (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) sample_with_burn_in() (synora.memory.iris_memory.IRISReplayBuffer method) SampleDist (class in synora.vision.dreamer_decoder) sampling_method (synora.configs.diamond_config.DiamondConfig attribute) save() (synora.models.dreamer.Dreamer method) (synora.models.iris_agent.IRISAgent method) save_bundle() (in module synora.inference.bundle) save_checkpoint() (in module synora.training.train_convvae) (in module synora.training.train_mdn_rnn) (synora.training.train_diamond.DiamondAgent method) (synora.training.train_genie.GenieTrainer method) save_frames() (in module synora.utils.utils) save_interval (synora.configs.diamond_config.DiamondConfig attribute) save_pretrained() (synora.models.diffusion.DiT.DiT method) (synora.models.genie.Genie method) save_video() (in module synora.utils.utils) scalar_freq (synora.configs.dreamer_config.DreamerConfig attribute) scale (synora.inference.quantize.Int8WeightOnlyLinear attribute) scenario_name (synora.datasets.nuplan.NuPlanSample attribute) scheduler_factor (synora.configs.wm_config.WMMDNRNNConfig attribute) (synora.configs.wm_config.WMVAEConfig attribute) scheduler_patience (synora.configs.wm_config.WMMDNRNNConfig attribute) (synora.configs.wm_config.WMVAEConfig attribute) seed (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.dreamer_config.DreamerConfig attribute) seed() (synora.envs.diamond_atari.DiamondAtariWrapper method) seed_steps (synora.configs.dreamer_config.DreamerConfig attribute) SelectAction (class in synora.envs.wrappers) SelfAttentionBlock (class in synora.vision.iris_encoder) seq_len (synora.configs.wm_config.WMMDNRNNConfig attribute) seq_to_batch() (synora.models.dreamer_rssm.RSSM method) (synora.models.modular_rssm.ModularRSSM method) SequenceDataset (class in synora.datasets.diamond_dataset) (class in synora.datasets.wm_dataset) setup_logging() (in module synora.utils.logging_utils) shuffle (synora.datasets.video_datasets.DatasetConfig attribute) sigma_data (synora.configs.diamond_config.DiamondConfig attribute) sigma_max (synora.configs.diamond_config.DiamondConfig attribute) sigma_min (synora.configs.diamond_config.DiamondConfig attribute) sign (synora.vision.dreamer_decoder.TanhBijector property) SimWorker (class in synora.envs.vector_env) sinusoidal_time_embedding() (in module synora.models.diffusion.DiT) size (synora.memory.planet_memory.Episode property) (synora.memory.planet_memory.Memory property) slave_routine() (in module synora.training.train_controller) split (synora.datasets.tinyworlds.TinyWorldsConfig attribute) start_actor_critic_after (synora.configs.iris_config.IRISConfig attribute) start_autoencoder_after (synora.configs.iris_config.IRISConfig attribute) start_lr (synora.configs.jepa_config.JEPAConfig attribute) start_transformer_after (synora.configs.iris_config.IRISConfig attribute) state (synora.inference.steppers.ImagineOutput attribute) state_dict() (synora.datasets.diamond_dataset.ReplayBuffer method) (synora.inference.bundle.Bundle method) (synora.utils.train_utils.EarlyStopping method) (synora.utils.train_utils.ReduceLROnPlateau method) state_posterior() (synora.models.rssm.RecurrentStateSpaceModel method) state_prior() (synora.models.rssm.RecurrentStateSpaceModel method) step() (synora.envs.diamond_atari.DiamondAtariWrapper method) (synora.envs.dmc.DeepMindControlEnv method) (synora.envs.dmlab.DMLabEnv method) (synora.envs.gym_env.GymImageEnv method) (synora.envs.mujoco_env.MuJoCoImageEnv method) (synora.envs.procgen_env.ProcgenImageEnv method) (synora.envs.unity_env.UnityMLAgentsEnv method) (synora.envs.wrappers.ActionRepeat method) (synora.envs.wrappers.FrameStack method) (synora.envs.wrappers.NormalizeActions method) (synora.envs.wrappers.ObsDict method) (synora.envs.wrappers.OneHotAction method) (synora.envs.wrappers.RenderImage method) (synora.envs.wrappers.ResizeImage method) (synora.envs.wrappers.RewardObs method) (synora.envs.wrappers.SelectAction method) (synora.envs.wrappers.TimeLimit method) (synora.masks.multiblock.MaskCollator method) (synora.masks.random.MaskCollator method) (synora.utils.jepa_utils.CosineWDSchedule method) (synora.utils.jepa_utils.WarmupCosineSchedule method) (synora.utils.train_utils.EarlyStopping method) (synora.utils.train_utils.ReduceLROnPlateau method) (synora.utils.utils.TorchImageEnvWrapper method) step_batch() (synora.envs.vector_env.TorchVectorizedEnv method) (synora.envs.vector_env.VectorizedEnv method) step_module() (synora.inference.steppers.DreamerStepper method) steps (synora.inference.benchmark.DriftReport attribute) steps_per_sec (synora.inference.benchmark.LatencyReport attribute) STKVCache (class in synora.blocks.st_transformer) STMLP (class in synora.blocks.st_transformer) stoch_size (synora.configs.dreamer_config.DreamerConfig attribute) (synora.models.modular_rssm.BackboneBase attribute) (synora.models.modular_rssm.ModularRSSM property) stop (synora.utils.train_utils.EarlyStopping property) StreamingVideoWriter (class in synora.utils.utils) STSpatialAttention (class in synora.blocks.st_transformer) STTemporalAttention (class in synora.blocks.st_transformer) STTransformer (class in synora.blocks.st_transformer) STTransformerBlock (class in synora.blocks.st_transformer) STTransformerConfig (class in synora.configs.genie_config) summary() (synora.inference.benchmark.DriftReport method) (synora.inference.benchmark.LatencyReport method) (synora.models.diffusion.DiT.DiT method) (synora.models.dreamer.Dreamer method) (synora.models.dreamer.DreamerAgent method) (synora.models.genie.Genie method) (synora.models.iris_agent.IRISAgent method) (synora.models.jepa_agent.JEPAAgent method) (synora.training.train_diamond.DiamondAgent method) supports_imagination (synora.inference.steppers.DreamerStepper attribute) (synora.inference.steppers.IRISStepper attribute) (synora.inference.steppers.WorldModelStepper attribute) sweep_config (synora.configs.jepa_config.JEPAConfig attribute) sweep_train() (in module synora.training.train_jepa) symexp() (in module synora.utils.dreamer_utils) symlog() (in module synora.utils.dreamer_utils) symlog_range (synora.configs.dreamer_config.DreamerConfig attribute) synora.api module synora.benchmarks module synora.benchmarks.adapters module synora.benchmarks.metrics module synora.benchmarks.reporting module synora.benchmarks.runner module synora.blocks module synora.blocks.mhsa module synora.blocks.st_transformer module synora.catalog module synora.configs.diamond_config module synora.configs.dit_config module synora.configs.dreamer_config module synora.configs.genie_config module synora.configs.iris_config module synora.configs.jepa_config module synora.configs.wm_config module synora.controller.iris_policy module synora.controller.rollout_generator module

synora.controller.rssm_policy module synora.datasets.cifar10 module synora.datasets.diamond_dataset module synora.datasets.imagenet1k module synora.datasets.nuplan module synora.datasets.tinyworlds module synora.datasets.video_datasets module synora.datasets.wm_dataset module synora.envs.ale_atari_env module synora.envs.ale_atari_vector_env module synora.envs.diamond_atari module synora.envs.dmc module synora.envs.dmlab module synora.envs.gym_env module synora.envs.mujoco_env module synora.envs.procgen_env module synora.envs.robotics_env module synora.envs.unity_env module synora.envs.vector_env module synora.envs.wrappers module synora.export module synora.helpers.jepa_helper module synora.inference.benchmark module synora.inference.bundle module synora.inference.precision module synora.inference.quantize module synora.inference.runtime module synora.inference.steppers module synora.losses.convae_loss module synora.losses.gmm_loss module synora.masks module synora.masks.default module synora.masks.multiblock module synora.masks.random module synora.memory.dreamer_memory module synora.memory.iris_memory module synora.memory.planet_memory module synora.models.controller module synora.models.diffusion module synora.models.diffusion.actor_critic module synora.models.diffusion.DDPM module synora.models.diffusion.diamond_diffusion module synora.models.diffusion.DiT module synora.models.diffusion.reward_termination module synora.models.dreamer module synora.models.dreamer_rssm module synora.models.dynamics_model module synora.models.genie module synora.models.iris_agent module synora.models.iris_transformer module synora.models.jepa_agent module synora.models.latent_action_model module synora.models.mdrnn module synora.models.modular_rssm module synora.models.planet module synora.models.rssm module synora.models.vit module synora.training module synora.training.rl_harness module synora.training.train_controller module synora.training.train_convvae module synora.training.train_diamond module synora.training.train_genie module synora.training.train_iris module synora.training.train_jepa module synora.training.train_mdn_rnn module synora.training.train_planet module synora.training.train_rssm module synora.training.train_world_model module synora.transforms.image module synora.utils.data_utils module synora.utils.dreamer_utils module synora.utils.jepa_utils module synora.utils.logging_utils module synora.utils.memory_utils module synora.utils.train_utils module synora.utils.utils module synora.vision.dreamer_decoder module synora.vision.dreamer_encoder module synora.vision.iris_decoder module synora.vision.iris_encoder module synora.vision.planet_decoder module synora.vision.planet_encoder module synora.vision.VAE.ConvVAE module synora.vision.video_tokenizer module synora.vision.vq_layer module

T

TanhBijector (class in synora.vision.dreamer_decoder) target_return (synora.configs.wm_config.WMControllerConfig attribute) td_lambda (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.iris_config.IRISConfig attribute) TensorBoardMetrics (class in synora.utils.utils) terminate() (synora.memory.planet_memory.Episode method) terminate_on_life_loss (synora.configs.diamond_config.DiamondConfig attribute) test (synora.configs.dreamer_config.DreamerConfig attribute) test_episodes (synora.configs.dreamer_config.DreamerConfig attribute) test_epoch() (in module synora.training.train_convvae) test_interval (synora.configs.dreamer_config.DreamerConfig attribute) test_trained_model() (in module synora.training.train_world_model) tf32 (synora.configs.dreamer_config.DreamerConfig attribute) time_limit (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.wm_config.WMControllerConfig attribute) TimeLimit (class in synora.envs.wrappers) TimestepEmbedder (class in synora.models.diffusion.DiT) TimestepEmbedding (class in synora.models.diffusion.diamond_diffusion) TIMESTEPS (synora.configs.dit_config.DiTConfig attribute) TinyWorldsConfig (class in synora.datasets.tinyworlds) TinyWorldsDataLoader (class in synora.datasets.tinyworlds) TinyWorldsDataset (class in synora.datasets.tinyworlds) to() (synora.utils.dreamer_utils.TwoHotEncoder method) to_channels_last() (in module synora.utils.memory_utils) to_dict() (synora.configs.jepa_config.JEPAConfig method) (synora.configs.wm_config.WMControllerConfig method) (synora.configs.wm_config.WMMDNRNNConfig method) (synora.configs.wm_config.WMVAEConfig method) to_float_tensor() (synora.training.train_iris.IRISTrainer method) to_latent() (in module synora.training.train_mdn_rnn) to_model_domain() (in module synora.datasets.diamond_dataset) to_nested_dict() (synora.configs.jepa_config.JEPAConfig method) to_pixel_domain() (in module synora.datasets.diamond_dataset) to_tensor_obs() (in module synora.utils.utils) to_train_dict() (synora.configs.jepa_config.JEPAConfig method) token_drop() (synora.models.diffusion.DiT.LabelEmbedder method) token_embedding_dim (synora.configs.iris_config.IRISConfig attribute) tokenizer_decoder_depth (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) tokenizer_decoder_dim (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) tokenizer_embedding_dim (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) tokenizer_encoder_depth (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) tokenizer_encoder_dim (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute)

tokenizer_num_heads (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) tokenizer_vocab_size (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) tokens_per_frame (synora.configs.iris_config.IRISConfig attribute) TorchImageEnvWrapper (class in synora.utils.utils) TorchVectorizedEnv (class in synora.envs.vector_env) total_epochs (synora.configs.iris_config.IRISConfig attribute) total_steps (synora.configs.dreamer_config.DreamerConfig attribute) train (synora.configs.dreamer_config.DreamerConfig attribute) train() (in module synora.training.train_planet) (synora.models.diffusion.DiT.DiT method) (synora.models.dreamer.DreamerAgent method) (synora.models.jepa_agent.JEPAAgent method) (synora.models.planet.Planet method) (synora.training.rl_harness.PPOTrainer method) (synora.training.train_diamond.DiamondAgent method) (synora.training.train_genie.GenieTrainer method) (synora.training.train_iris.IRISTrainer method) train_batch_size (synora.configs.wm_config.WMVAEConfig attribute) train_controller() (in module synora.training.train_controller) train_convae() (in module synora.training.train_convvae) train_diamond() (in module synora.training.train_diamond) train_epoch() (in module synora.training.train_convvae) (synora.training.train_iris.IRISTrainer method) train_mdn_rnn() (in module synora.training.train_mdn_rnn) train_one_batch() (synora.models.dreamer.Dreamer method) train_rssm() (in module synora.training.train_rssm) train_seq_len (synora.configs.dreamer_config.DreamerConfig attribute) train_step() (synora.training.rl_harness.PPOTrainer method) (synora.training.train_genie.GenieTrainer method) training_step() (synora.models.genie.Genie method) training_steps_per_epoch (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.iris_config.IRISConfig attribute) transform_reward() (synora.models.iris_agent.IRISAgent method) transformer_batch_size (synora.configs.iris_config.IRISConfig attribute) transformer_dropout (synora.configs.iris_config.IRISConfig attribute) transformer_embed_dim (synora.configs.iris_config.IRISConfig attribute) transformer_heads (synora.configs.iris_config.IRISConfig attribute) transformer_layers (synora.configs.iris_config.IRISConfig attribute) transformer_steps_per_epoch (synora.configs.iris_config.IRISConfig attribute) transformer_timesteps (synora.configs.iris_config.IRISConfig attribute) TransformerBackbone (class in synora.models.modular_rssm) TransformerBlock (class in synora.models.diffusion.DiT) (class in synora.models.modular_rssm) trunc_normal_() (in module synora.utils.jepa_utils) TwoHotEncoder (class in synora.utils.dreamer_utils)

U

unity_base_port (synora.configs.dreamer_config.DreamerConfig attribute) unity_behavior_name (synora.configs.dreamer_config.DreamerConfig attribute) unity_file_name (synora.configs.dreamer_config.DreamerConfig attribute) unity_no_graphics (synora.configs.dreamer_config.DreamerConfig attribute) unity_quality_level (synora.configs.dreamer_config.DreamerConfig attribute) unity_time_scale (synora.configs.dreamer_config.DreamerConfig attribute) unity_worker_id (synora.configs.dreamer_config.DreamerConfig attribute) UnityMLAgentsEnv (class in synora.envs.unity_env) unpatchify_tokens() (synora.models.diffusion.DiT.DiT method) UpBlock (class in synora.models.diffusion.diamond_diffusion) update() (synora.utils.jepa_utils.AverageMeter method) (synora.utils.utils.TensorBoardMetrics method) update_actor_critic() (synora.models.iris_agent.IRISAgent method) update_autoencoder() (synora.models.iris_agent.IRISAgent method)

update_steps (synora.configs.dreamer_config.DreamerConfig attribute) update_transformer() (synora.models.iris_agent.IRISAgent method) UpsampleBlock (class in synora.vision.iris_decoder) use_amp (synora.configs.diamond_config.DiamondConfig attribute) (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.configs.iris_config.IRISConfig attribute) use_bfloat16 (synora.configs.jepa_config.JEPAConfig attribute) use_color_distortion (synora.configs.jepa_config.JEPAConfig attribute) use_disc_model (synora.configs.dreamer_config.DreamerConfig attribute) use_ema (synora.configs.genie_config.VideoTokenizerConfig attribute) use_gaussian_blur (synora.configs.jepa_config.JEPAConfig attribute) use_horizontal_flip (synora.configs.jepa_config.JEPAConfig attribute) UUID (class in synora.envs.wrappers)

V

VAL_SPLIT (synora.configs.dit_config.DiTConfig attribute) val_split (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.configs.jepa_config.JEPAConfig attribute) (synora.training.train_genie.GenieConfig attribute) validate() (synora.configs.wm_config.WMControllerConfig method) (synora.configs.wm_config.WMMDNRNNConfig method) (synora.configs.wm_config.WMVAEConfig method) (synora.training.train_genie.GenieTrainer method) validate_epoch() (synora.training.train_genie.GenieTrainer method) value_learning_rate (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.iris_config.IRISConfig attribute) value_loss() (synora.models.diffusion.actor_critic.RLLoss method) (synora.models.dreamer.Dreamer method) VectorizedEnv (class in synora.envs.vector_env) VectorQuantizer (class in synora.vision.vq_layer) VectorQuantizerEMA (class in synora.vision.vq_layer) verify_export() (in module synora.export) video_format (synora.configs.dreamer_config.DreamerConfig attribute) video_fps (synora.configs.dreamer_config.DreamerConfig attribute) video_paths (synora.datasets.video_datasets.VideoDatasetBase attribute)

VideoDataset (class in synora.training.train_genie) VideoDatasetBase (class in synora.datasets.video_datasets) VideoDatasetConfig (class in synora.datasets.video_datasets) VideoFolderDataset (class in synora.datasets.video_datasets) VideoTokenizer (class in synora.vision.video_tokenizer) VideoTokenizerConfig (class in synora.configs.genie_config) VisionTransformer (class in synora.models.vit) VisionTransformerPredictor (class in synora.models.vit) visualize_latent_tsne() (in module synora.utils.utils) visualize_latent_umap() (in module synora.utils.utils) vit_base() (in module synora.models.vit) vit_giant() (in module synora.models.vit) vit_huge() (in module synora.models.vit) vit_large() (in module synora.models.vit) vit_predictor() (in module synora.models.vit) vit_small() (in module synora.models.vit) vit_tiny() (in module synora.models.vit) ViTEncoder (class in synora.models.modular_rssm) vocab_size (synora.configs.genie_config.DynamicsModelConfig attribute) (synora.configs.genie_config.LatentActionModelConfig attribute) (synora.configs.genie_config.VideoTokenizerConfig attribute) (synora.configs.iris_config.IRISConfig attribute)

W

wandb_entity (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.jepa_config.JEPAConfig attribute) wandb_project (synora.configs.dreamer_config.DreamerConfig attribute) (synora.configs.jepa_config.JEPAConfig attribute) warmup (synora.configs.jepa_config.JEPAConfig attribute) warmup() (synora.models.planet.Planet method) warmup_steps (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.training.train_genie.GenieConfig attribute) WarmupCosineSchedule (class in synora.utils.jepa_utils) WEIGHT_DECAY (synora.configs.dit_config.DiTConfig attribute) weight_decay (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.configs.iris_config.IRISConfig attribute) (synora.configs.jepa_config.JEPAConfig attribute) (synora.training.train_genie.GenieConfig attribute) weight_decay_actor (synora.configs.diamond_config.DiamondConfig attribute) weight_decay_diffusion (synora.configs.diamond_config.DiamondConfig attribute) weight_decay_reward (synora.configs.diamond_config.DiamondConfig attribute)

weight_int8 (synora.inference.quantize.Int8WeightOnlyLinear attribute) weight_memory_bytes() (in module synora.inference.quantize) WIDTH (synora.configs.dit_config.DiTConfig attribute) width (synora.configs.wm_config.WMVAEConfig attribute) window_attention_heads (synora.configs.genie_config.GenieConfig attribute) (synora.configs.genie_config.GenieSmallConfig attribute) (synora.configs.genie_config.LatentActionModelConfig attribute) (synora.training.train_genie.GenieConfig attribute) within() (synora.inference.benchmark.DriftReport method) WMControllerConfig (class in synora.configs.wm_config) WMMDNRNNConfig (class in synora.configs.wm_config) WMVAEConfig (class in synora.configs.wm_config) WORKDIR (synora.configs.dit_config.DiTConfig attribute) world_model_loss() (synora.models.dreamer.Dreamer method) WorldModelStepper (class in synora.inference.steppers) worst_rel_error (synora.inference.benchmark.DriftReport property) write() (synora.utils.utils.Mp4Writer method) write_frame() (synora.utils.utils.StreamingVideoWriter method) write_tag (synora.configs.jepa_config.JEPAConfig attribute)