Index A | B | C | D | E | F | G | I | K | L | M | N | O | P | R | S | T | U | V | W A AdaLNGenerator (class in autocast.processors.swin_vit) AdvectionDiffusionDataset (class in autocast.data.dataset) AE (class in autocast.models.autoencoder) AELoss (class in autocast.models.autoencoder) AlphaFairCRPS (class in autocast.metrics.ensemble) AlphaFairCRPSLoss (class in autocast.losses.ensemble) AlphaFairCRPSMAETerm (class in autocast.metrics.ensemble) AlphaFairCRPSSpreadTerm (class in autocast.metrics.ensemble) apply_gate() (in module autocast.processors.swin_vit) autocast.benchmarking.inference module autocast.data.datamodule module autocast.data.dataset module autocast.data.encoded_dataset module autocast.data.metadata module autocast.decoders.base module autocast.decoders.dc module autocast.decoders.identity module autocast.encoders.base module autocast.encoders.dc module autocast.encoders.identity module autocast.encoders.unet module autocast.logging.wandb module autocast.losses.ensemble module autocast.metrics.base module autocast.metrics.coverage module autocast.metrics.deterministic module autocast.metrics.ensemble module autocast.models.autoencoder module autocast.models.encoder_decoder module autocast.models.encoder_processor_decoder module autocast.models.processor module autocast.models.variational_autoencoder module autocast.nn.residual module autocast.nn.spatial_attention module autocast.processors.base module autocast.processors.diffusion module autocast.processors.flow_matching module autocast.processors.noise module autocast.processors.swin_vit module autocast.processors.unet module autocast.processors.vit module autocast.types.batch module autocast.types.types module autocast.utils.optimizer module autocast.utils.plots module AViT (class in autocast.processors.vit) AViTProcessor (class in autocast.processors.vit) AxialAttentionBlock (class in autocast.processors.vit) AzulaUNetProcessor (class in autocast.processors.unet) B BaseMetric (class in autocast.metrics.base) BasicSwinLayer (class in autocast.processors.swin_vit) Batch (class in autocast.types.batch) BatchMixin (class in autocast.data.dataset) benchmark_model() (in module autocast.benchmarking.inference) benchmark_rollout() (in module autocast.benchmarking.inference) boundary_condition_types (Metadata attribute) boundary_conditions (Batch attribute) (Sample attribute) BOUTDataset (class in autocast.data.dataset) BTSCMetric (class in autocast.metrics.deterministic) BTSCMMetric (class in autocast.metrics.ensemble) C CachedLatentDataset (class in autocast.data.encoded_dataset) channel_axis (DCEncoder attribute) (Encoder attribute) (IdentityEncoder attribute) compute() (BaseMetric method) (MultiCoverage method) (MultiWinkler method) compute_coverage_scores_from_dataloader() (in module autocast.utils.plots) compute_detailed() (MultiCoverage method) compute_metrics_from_dataloader() (in module autocast.utils.plots) compute_metrics_per_timestep_from_dataloader() (in module autocast.utils.plots) constant_field_names (Metadata attribute) constant_fields (Batch attribute) (Sample attribute) constant_scalar_names (Metadata attribute) constant_scalars (Batch attribute) (Sample attribute) Coverage (class in autocast.metrics.coverage) create_notebook_logger() (in module autocast.logging.wandb) create_wandb_logger() (in module autocast.logging.wandb) CRPS (class in autocast.metrics.ensemble) CRPSLoss (class in autocast.losses.ensemble) CRPSMAETerm (class in autocast.metrics.ensemble) CRPSSpreadTerm (class in autocast.metrics.ensemble) D dataset_name (Metadata attribute) DCDecoder (class in autocast.decoders.dc) DCEncoder (class in autocast.encoders.dc) decode() (DCDecoder method) (Decoder method) (EncoderDecoder method) (IdentityDecoder method) decoder (AE attribute) Decoder (class in autocast.decoders.base) decoder (EncoderDecoder attribute) (VAE attribute) decoder_model (DCDecoder attribute) DiffusionProcessor (class in autocast.processors.diffusion) DiscreteProcessor (class in autocast.models.processor) E encode() (DCEncoder method) (Encoder method) (EncoderDecoder method) (GenericEncoder method) (IdentityEncoder method) (UNetEncoder method) (VAE method) encode_batch() (EncoderWithCond method) (GenericEncoder method) encode_cond() (EncoderWithCond method) encode_tensor() (DCEncoder method) encode_with_cond() (EncoderWithCond method) encoded_info (EncodedBatch attribute) (EncodedSample attribute) encoded_inputs (EncodedBatch attribute) (EncodedSample attribute) encoded_output_fields (EncodedBatch attribute) (EncodedSample attribute) EncodedBatch (class in autocast.types.batch) EncodedBatchMixin (class in autocast.data.encoded_dataset) EncodedDataModule (class in autocast.data.encoded_dataset) EncodedDataset (class in autocast.data.encoded_dataset) EncodedSample (class in autocast.types.batch) encoder (AE attribute) Encoder (class in autocast.encoders.base) encoder (EncoderDecoder attribute) (VAE attribute) encoder_decoder (EncoderProcessorDecoder attribute) encoder_model (DCEncoder attribute) (Encoder attribute) (UNetEncoder attribute) EncoderDecoder (class in autocast.models.encoder_decoder) EncoderProcessorDecoder (class in autocast.models.encoder_processor_decoder) EncoderWithCond (class in autocast.encoders.base) EnergyScore (class in autocast.metrics.ensemble) EnsembleLoss (class in autocast.losses.ensemble) EnsembleMAELoss (class in autocast.losses.ensemble) EnsembleSkill (class in autocast.metrics.ensemble) EnsembleSpread (class in autocast.metrics.ensemble) F FairCRPS (class in autocast.metrics.ensemble) FairCRPSLoss (class in autocast.losses.ensemble) FairCRPSMAETerm (class in autocast.metrics.ensemble) FairCRPSSpreadTerm (class in autocast.metrics.ensemble) fc_log_var (VAE attribute) fc_mean (VAE attribute) field_names (Metadata attribute) flow_field() (FlowMatchingProcessor method) FlowBasedGenerativeProcessor (class in autocast.models.processor) FlowMatchingProcessor (class in autocast.processors.flow_matching) forward() (AdaLNGenerator method) (AE method) (AELoss method) (AViT method) (AViTProcessor method) (AxialAttentionBlock method) (AzulaUNetProcessor method) (BasicSwinLayer method) (Decoder method) (DiffusionProcessor method) (EncoderDecoder method) (EncoderProcessorDecoder method) (EncoderWithCond method) (EnsembleLoss method) (FlowMatchingProcessor method) (GenericEncoder method) (LogLinearSchedule method) (NoiseSchedule method) (PatchEmbedding method) (PatchMerging method) (PatchSplitting method) (PatchUnembedding method) (ProcessorModel method) (ResBlock method) (Residual method), [1] (SpatialAttentionWrapper method), [1] (SwinViTBlock method) (SwinViTProcessor method) (UNetClassic method) (UNetProcessor method) (VAE method), [1] (VAELoss method) (VMSELoss method) forward_with_latent() (AE method) (EncoderDecoder method) (VAE method) from_files() (MiniWellDataset static method) (MiniWellInputOutput static method) G GenericEncoder (class in autocast.encoders.base) get_encoder_specs() (SwinViTProcessor method) get_loss() (AELoss static method) get_optimizer_config() (in module autocast.utils.optimizer) global_cond (EncodedBatch attribute) (EncodedSample attribute) grid_type (Metadata attribute) I IdentityDecoder (class in autocast.decoders.identity) IdentityEncoder (class in autocast.encoders.identity) input_fields (Batch attribute) (Sample attribute) K kl_divergence() (VAELoss method) L latent_channels (Decoder attribute) (Encoder attribute) learning_rate (Processor attribute) LInfinity (class in autocast.metrics.deterministic) log_metrics() (in module autocast.logging.wandb) LogLinearSchedule (class in autocast.processors.noise) loss() (AViTProcessor method) (AzulaUNetProcessor method) (DiffusionProcessor method) (EncoderProcessorDecoder method) (FlowMatchingProcessor method) (Processor method) (ProcessorModel method) (SwinViTProcessor method) (UNetProcessor method) loss_func (EncoderDecoder attribute) M MAE (class in autocast.metrics.deterministic) make_synthetic_batch() (in module autocast.benchmarking.inference) map() (AViTProcessor method) (AzulaUNetProcessor method) (DiffusionProcessor method) (FlowMatchingProcessor method) (Processor method) (ProcessorModel method) (SwinViTProcessor method) (UNetProcessor method) maybe_watch_model() (in module autocast.logging.wandb) measure_flops() (in module autocast.benchmarking.inference) Metadata (class in autocast.data.metadata) miniwell_dataset (MiniWellInputOutput attribute) MiniWellDataModule (class in autocast.data.encoded_dataset) MiniWellDataset (class in autocast.data.encoded_dataset) MiniWellInputOutput (class in autocast.data.encoded_dataset) modulate() (in module autocast.processors.swin_vit) module autocast.benchmarking.inference autocast.data.datamodule autocast.data.dataset autocast.data.encoded_dataset autocast.data.metadata autocast.decoders.base autocast.decoders.dc autocast.decoders.identity autocast.encoders.base autocast.encoders.dc autocast.encoders.identity autocast.encoders.unet autocast.logging.wandb autocast.losses.ensemble autocast.metrics.base autocast.metrics.coverage autocast.metrics.deterministic autocast.metrics.ensemble autocast.models.autoencoder autocast.models.encoder_decoder autocast.models.encoder_processor_decoder autocast.models.processor autocast.models.variational_autoencoder autocast.nn.residual autocast.nn.spatial_attention autocast.processors.base autocast.processors.diffusion autocast.processors.flow_matching autocast.processors.noise autocast.processors.swin_vit autocast.processors.unet autocast.processors.vit autocast.types.batch autocast.types.types autocast.utils.optimizer autocast.utils.plots MSE (class in autocast.metrics.deterministic) MultiCoverage (class in autocast.metrics.coverage) MultiWinkler (class in autocast.metrics.ensemble) N n_files (Metadata attribute) n_spatial_dims (Metadata attribute) n_steps_per_trajectory (Metadata attribute) n_trajectories_per_file (Metadata attribute) name (AlphaFairCRPS attribute) (AlphaFairCRPSMAETerm attribute) (AlphaFairCRPSSpreadTerm attribute) (Coverage attribute) (CRPS attribute) (CRPSMAETerm attribute) (CRPSSpreadTerm attribute) (EnergyScore attribute) (EnsembleSkill attribute) (EnsembleSpread attribute) (FairCRPS attribute) (FairCRPSMAETerm attribute) (FairCRPSSpreadTerm attribute) (LInfinity attribute) (MAE attribute) (MSE attribute) (MultiWinkler attribute) (NMAE attribute) (NMSE attribute) (NRMSE attribute) (PowerSpectrumCCRMSE attribute) (PowerSpectrumCCRMSEHigh attribute) (PowerSpectrumCCRMSELow attribute) (PowerSpectrumCCRMSEMid attribute) (PowerSpectrumCCRMSETail attribute) (PowerSpectrumRMSE attribute) (PowerSpectrumRMSEHigh attribute) (PowerSpectrumRMSELow attribute) (PowerSpectrumRMSEMid attribute) (PowerSpectrumRMSETail attribute) (RMSE attribute) (SpreadSkillRatio attribute) (VariogramScore attribute) (VMSE attribute) (VRMSE attribute) (WinklerScore attribute) NMAE (class in autocast.metrics.deterministic) NMSE (class in autocast.metrics.deterministic) NoiseSchedule (class in autocast.processors.noise) NRMSE (class in autocast.metrics.deterministic) O on_load_checkpoint() (EncoderProcessorDecoder method) optimizer_config (EncoderDecoder attribute) (ProcessorModel attribute) optional_checkpointing() (UNetClassic method) output_fields (Batch attribute) (Sample attribute) outputs_time_channel_concat (Encoder attribute) P parse_data() (SpatioTemporalDataset method) PatchEmbedding (class in autocast.processors.vit) PatchMerging (class in autocast.processors.swin_vit) PatchSplitting (class in autocast.processors.swin_vit) PatchUnembedding (class in autocast.processors.vit) plot() (MultiCoverage method) (MultiWinkler method) plot_coverage() (in module autocast.utils.plots) plot_spatiotemporal_snapshots() (in module autocast.utils.plots) plot_spatiotemporal_snapshots_data_only() (in module autocast.utils.plots) plot_spatiotemporal_video() (in module autocast.utils.plots) postprocess() (Decoder method) PowerSpectrumCCRMSE (class in autocast.metrics.deterministic) PowerSpectrumCCRMSEHigh (class in autocast.metrics.deterministic) PowerSpectrumCCRMSELow (class in autocast.metrics.deterministic) PowerSpectrumCCRMSEMid (class in autocast.metrics.deterministic) PowerSpectrumCCRMSETail (class in autocast.metrics.deterministic) PowerSpectrumRMSE (class in autocast.metrics.deterministic) PowerSpectrumRMSEHigh (class in autocast.metrics.deterministic) PowerSpectrumRMSELow (class in autocast.metrics.deterministic) PowerSpectrumRMSEMid (class in autocast.metrics.deterministic) PowerSpectrumRMSETail (class in autocast.metrics.deterministic) preprocess() (GenericEncoder method) Processor (class in autocast.processors.base) processor (EncoderProcessorDecoder attribute) (ProcessorModel attribute) ProcessorModel (class in autocast.models.processor) R ReactionDiffusionDataset (class in autocast.data.dataset) read_data() (SpatioTemporalDataset method) reparametrize() (VAE method), [1] repeat() (Batch method) (EncodedBatch method) ResBlock (class in autocast.nn.residual) reset() (BaseMetric method) (MultiCoverage method) (MultiWinkler method) Residual (class in autocast.nn.residual) (class in autocast.nn.spatial_attention) RMSE (class in autocast.metrics.deterministic) rollout_test_dataloader() (EncodedDataModule method) (SpatioTemporalDataModule method) (TheWellDataModule method) rollout_val_dataloader() (SpatioTemporalDataModule method) (TheWellDataModule method) S Sample (class in autocast.types.batch) sample() (DiffusionProcessor method) scalar_names (Metadata attribute) score() (BaseMetric method) (BTSCMMetric method) (EnsembleSkill method) (EnsembleSpread method) (MultiWinkler method) (SpreadSkillRatio method) set_up_normalization() (SpatioTemporalDataset method) setup() (EncodedDataModule method) (MiniWellDataModule method) spatial_resolution (Metadata attribute) SpatialAttentionWrapper (class in autocast.nn.residual) (class in autocast.nn.spatial_attention) SpatioTemporalDataModule (class in autocast.data.datamodule) SpatioTemporalDataset (class in autocast.data.dataset) SpreadSkillRatio (class in autocast.metrics.ensemble) SwinViTBlock (class in autocast.processors.swin_vit) SwinViTProcessor (class in autocast.processors.swin_vit) T test_dataloader() (EncodedDataModule method) (MiniWellDataModule method) (SpatioTemporalDataModule method) (TheWellDataModule method) test_metrics (EncoderProcessorDecoder attribute) test_step() (AE method) (EncoderDecoder method) (EncoderProcessorDecoder method) (ProcessorModel method) TheWell (class in autocast.data.dataset) TheWellDataModule (class in autocast.data.datamodule) to() (Batch method) (EncodedBatch method) to_preloaded_data() (SpatioTemporalDataset method) to_sample() (BatchMixin static method) (EncodedBatchMixin static method) train_dataloader() (EncodedDataModule method) (MiniWellDataModule method) (SpatioTemporalDataModule method) (TheWellDataModule method) train_metrics (EncoderProcessorDecoder attribute) training_step() (AE method) (EncoderDecoder method) (EncoderProcessorDecoder method) (ProcessorModel method) (VAE method) U UNetClassic (class in autocast.processors.unet) UNetEncoder (class in autocast.encoders.unet) UNetProcessor (class in autocast.processors.unet) update() (BaseMetric method) (MultiCoverage method) (MultiWinkler method) V VAE (class in autocast.models.encoder_decoder) (class in autocast.models.variational_autoencoder) VAELoss (class in autocast.models.variational_autoencoder) val_dataloader() (EncodedDataModule method) (MiniWellDataModule method) (SpatioTemporalDataModule method) (TheWellDataModule method) val_metrics (EncoderProcessorDecoder attribute) validation_step() (AE method) (EncoderDecoder method) (EncoderProcessorDecoder method) (ProcessorModel method) VariogramScore (class in autocast.metrics.ensemble) vector_dims (EnergyScore attribute) (VariogramScore attribute) VMSE (class in autocast.metrics.deterministic) VMSELoss (class in autocast.models.autoencoder) VRMSE (class in autocast.metrics.deterministic) W well_dataset (TheWell attribute) WinklerScore (class in autocast.metrics.ensemble)
