Index

A

accelerator (anemoi.training.schemas.system.HardwareSchema attribute) accumulation_levels_plot (anemoi.training.schemas.diagnostics.PlotEnsSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotSampleSchema attribute) add_scaler() (anemoi.training.losses.base.BaseLoss method) (anemoi.training.losses.base.BaseLossWrapper method) (anemoi.training.losses.combined.CombinedLoss method) AdEMAMix (class in anemoi.training.optimizers.AdEMAMix) advance_input() (anemoi.training.tasks.base.BaseSingleStepTask method) allgather_batch() (anemoi.training.train.methods.base.BaseTrainingModule method) anemoi.graphs.schemas.base_graph module anemoi.training.data.multidataset module anemoi.training.diagnostics.callbacks.checkpoint module anemoi.training.diagnostics.callbacks.evaluation module anemoi.training.diagnostics.callbacks.optimiser module anemoi.training.diagnostics.callbacks.plot module anemoi.training.diagnostics.callbacks.plot_adapter module anemoi.training.diagnostics.callbacks.provenance module anemoi.training.losses.base module anemoi.training.losses.combined module anemoi.training.losses.utils module anemoi.training.optimizers.AdEMAMix module anemoi.training.schemas.data module

anemoi.training.schemas.diagnostics module anemoi.training.schemas.system module anemoi.training.tasks.base module anemoi.training.tasks.timeless module anemoi.training.train.methods.base module anemoi.training.train.methods.edm_diffusion module anemoi.training.train.methods.ensemble module anemoi.training.train.methods.single module anemoi.training.train.methods.stochastic_interpolant module anemoi.training.train.methods.transport module AnemoiCheckpoint (class in anemoi.training.diagnostics.callbacks.checkpoint) anomaly_detection (anemoi.training.schemas.diagnostics.Debug attribute) artifact_subfolder (anemoi.training.diagnostics.callbacks.plot.BasePlotCallback property) asynchronous (anemoi.training.schemas.diagnostics.PlotSchema attribute) AsyncPlotExecutor (class in anemoi.training.diagnostics.callbacks.plot) attributes (anemoi.graphs.schemas.base_graph.EdgeSchema attribute) (anemoi.graphs.schemas.base_graph.NodeSchema attribute) authentication (anemoi.training.schemas.diagnostics.MlflowSchema attribute) Autoencoder (class in anemoi.training.tasks.timeless) AutoencoderPlotAdapter (class in anemoi.training.diagnostics.callbacks.plot_adapter) auxiliary_label (anemoi.training.schemas.diagnostics.PlotSampleSchema attribute) azure_log_level (anemoi.training.schemas.diagnostics.AzureMlflowSchema attribute) AzureMlflowSchema (class in anemoi.training.schemas.diagnostics)

B

BaseGraphSchema (class in anemoi.graphs.schemas.base_graph) BaseLoss (class in anemoi.training.losses.base) BaseLossWrapper (class in anemoi.training.losses.base) BasePerBatchPlotCallback (class in anemoi.training.diagnostics.callbacks.plot) BasePerEpochPlotCallback (class in anemoi.training.diagnostics.callbacks.plot) BasePlotAdapter (class in anemoi.training.diagnostics.callbacks.plot_adapter) BasePlotAdditionalMetrics (class in anemoi.training.diagnostics.callbacks.plot) BasePlotCallback (class in anemoi.training.diagnostics.callbacks.plot)

BasePlotExecutor (class in anemoi.training.diagnostics.callbacks.plot) BaseSingleStepTask (class in anemoi.training.tasks.base) BaseTask (class in anemoi.training.tasks.base) BaseTimelessTask (class in anemoi.training.tasks.timeless) BaseTrainingModule (class in anemoi.training.train.methods.base) BaseTransportTraining (class in anemoi.training.train.methods.transport) batch (anemoi.training.schemas.diagnostics.PlottingFrequency attribute) benchmark_profiler (anemoi.training.schemas.diagnostics.DiagnosticsSchema attribute) BenchmarkProfilerSchema (class in anemoi.training.schemas.diagnostics)

C

calculate_difference() (anemoi.training.losses.base.FunctionalLoss method) calculate_val_metrics() (anemoi.training.train.methods.base.BaseTrainingModule method) callbacks (anemoi.training.schemas.diagnostics.DiagnosticsSchema attribute) (anemoi.training.schemas.diagnostics.PlotSchema attribute) check_finite (anemoi.training.schemas.diagnostics.EarlyStoppingSchema attribute) check_loss_tree_variable_units() (in module anemoi.training.losses.utils) check_on_train_epoch_end (anemoi.training.schemas.diagnostics.EarlyStoppingSchema attribute) check_val_every_n_epoch (anemoi.training.schemas.diagnostics.DiagnosticsSchema attribute) checkpoint (anemoi.training.schemas.diagnostics.DiagnosticsSchema attribute) checkpoints (anemoi.training.schemas.system.OutputSchema attribute) CheckpointSchema (class in anemoi.training.schemas.diagnostics) CheckpointsSchema (class in anemoi.training.schemas.system)

clevels (anemoi.training.schemas.diagnostics.MatplotlibColormapClevelsSchema attribute) colorblind_type (anemoi.training.schemas.diagnostics.DistinctipyColormapSchema attribute) colormaps (anemoi.training.schemas.diagnostics.PlotEnsSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotSchema attribute) CombinedLoss (class in anemoi.training.losses.combined) compute_dataset_loss_metrics() (anemoi.training.train.methods.base.BaseTrainingModule method) (anemoi.training.train.methods.ensemble.EnsembleTraining method) (anemoi.training.train.methods.transport.BaseTransportTraining method) compute_loss() (anemoi.training.train.methods.edm_diffusion.EDMDiffusionTransportObjective method) compute_loss_metrics() (anemoi.training.train.methods.base.BaseTrainingModule method) configure_optimizers() (anemoi.training.train.methods.base.BaseTrainingModule method)

D

data (anemoi.training.data.multidataset.MultiDataset property) DataSchema (class in anemoi.training.schemas.data) dataset (anemoi.training.schemas.system.InputSchema attribute) dataset_names (anemoi.training.schemas.diagnostics.GraphTrainableFeaturesPlotSchema attribute) (anemoi.training.schemas.diagnostics.PlotEnsSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotHistogramSchema attribute) (anemoi.training.schemas.diagnostics.PlotLossSchema attribute) (anemoi.training.schemas.diagnostics.PlotSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotSpectrumSchema attribute) DatasetDataSchema (class in anemoi.training.schemas.data)

datasets (anemoi.training.schemas.data.DataSchema attribute), [1] datasets_to_plot (anemoi.training.schemas.diagnostics.PlotSchema attribute) datashader (anemoi.training.schemas.diagnostics.PlotSchema attribute) DDPEnsGroupStrategy (class in anemoi.training.distributed.strategy) DDPGroupStrategy (class in anemoi.training.distributed.strategy) debug (anemoi.training.schemas.diagnostics.DiagnosticsSchema attribute) Debug (class in anemoi.training.schemas.diagnostics) diagnostic (anemoi.training.schemas.data.DatasetDataSchema attribute) DiagnosticsSchema (class in anemoi.training.schemas.diagnostics) DistinctipyColormapSchema (class in anemoi.training.schemas.diagnostics) divergence_threshold (anemoi.training.schemas.diagnostics.EarlyStoppingSchema attribute)

E

EarlyStoppingSchema (class in anemoi.training.schemas.diagnostics) edge_builders (anemoi.graphs.schemas.base_graph.EdgeSchema attribute) edges (anemoi.graphs.schemas.base_graph.BaseGraphSchema attribute) EdgeSchema (class in anemoi.graphs.schemas.base_graph) EDMDiffusionTransportObjective (class in anemoi.training.train.methods.edm_diffusion) enable_checkpointing (anemoi.training.schemas.diagnostics.DiagnosticsSchema attribute) enable_progress_bar (anemoi.training.schemas.diagnostics.DiagnosticsSchema attribute) enabled (anemoi.training.schemas.diagnostics.MemorySchema attribute) (anemoi.training.schemas.diagnostics.MlflowSchema attribute) (anemoi.training.schemas.diagnostics.Profiling attribute) (anemoi.training.schemas.diagnostics.Snapshot attribute) (anemoi.training.schemas.diagnostics.WandbSchema attribute) EnsemblePlotAdapterWrapper (class in anemoi.training.diagnostics.callbacks.plot_adapter) EnsembleTraining (class in anemoi.training.train.methods.ensemble)

entity (anemoi.training.schemas.diagnostics.WandbSchema attribute) epoch (anemoi.training.schemas.diagnostics.PlottingFrequency attribute) every_n_batches (anemoi.training.schemas.diagnostics.PlotEnsSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotHistogramSchema attribute) (anemoi.training.schemas.diagnostics.PlotLossSchema attribute) (anemoi.training.schemas.diagnostics.PlotSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotSpectrumSchema attribute) every_n_epochs (anemoi.training.schemas.diagnostics.GraphTrainableFeaturesPlotSchema attribute) (anemoi.training.schemas.system.CheckpointsSchema attribute) every_n_minutes (anemoi.training.schemas.system.CheckpointsSchema attribute) every_n_train_steps (anemoi.training.schemas.system.CheckpointsSchema attribute) expand_hyperparams (anemoi.training.schemas.diagnostics.MlflowSchema attribute) experiment_name (anemoi.training.schemas.diagnostics.MlflowSchema attribute) extra_plots (anemoi.training.schemas.diagnostics.MemorySchema attribute)

F

fill_metadata() (anemoi.training.tasks.base.BaseTask method) focus_area (anemoi.training.schemas.diagnostics.PlotEnsSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotHistogramSchema attribute) (anemoi.training.schemas.diagnostics.PlotSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotSpectrumSchema attribute) focus_areas (anemoi.training.schemas.diagnostics.PlotSchema attribute) FocusAreaSchema (class in anemoi.training.schemas.diagnostics) forcing (anemoi.training.schemas.data.DatasetDataSchema attribute) ForecasterPlotAdapter (class in anemoi.training.diagnostics.callbacks.plot_adapter) format (anemoi.training.schemas.data.DataSchema attribute), [1]

forward() (anemoi.training.losses.base.BaseLoss method) (anemoi.training.losses.base.FunctionalLoss method) (anemoi.training.losses.combined.CombinedLoss method) (anemoi.training.train.methods.base.BaseTrainingModule method) (anemoi.training.train.methods.ensemble.EnsembleTraining method) (anemoi.training.train.methods.transport.TransportTraining method) frequency (anemoi.training.data.multidataset.MultiDataset property) (anemoi.training.schemas.data.DataSchema attribute), [1] from_plot_config() (anemoi.training.diagnostics.callbacks.plot.PlottingSettings class method) FunctionalLoss (class in anemoi.training.losses.base)

G

get_batch_input_indices() (anemoi.training.tasks.base.BaseTask method) get_batch_output_indices() (anemoi.training.tasks.base.BaseTask method) get_data_output_target() (anemoi.training.train.methods.transport.BaseTransportTraining method) get_input_offsets() (anemoi.training.tasks.base.BaseTask method), [1] get_inputs() (anemoi.training.tasks.base.BaseTask method) get_metric_name() (anemoi.training.tasks.base.BaseTask method) get_offsets() (anemoi.training.tasks.base.BaseTask method), [1]

get_output_offsets() (anemoi.training.tasks.base.BaseTask method), [1] get_shard_slice() (anemoi.training.data.multidataset.MultiDataset method) get_targets() (anemoi.training.tasks.base.BaseTask method) gradients (anemoi.training.schemas.diagnostics.WandbSchema attribute) graph (anemoi.training.schemas.system.InputSchema attribute) GraphTrainableFeaturesPlot (class in anemoi.training.diagnostics.callbacks.plot) GraphTrainableFeaturesPlotSchema (class in anemoi.training.schemas.diagnostics)

H

hardware (anemoi.training.schemas.system.SystemSchema attribute) HardwareSchema (class in anemoi.training.schemas.system) has_scaler_for_dim() (anemoi.training.losses.base.BaseLoss method) (anemoi.training.losses.base.BaseLossWrapper method) (anemoi.training.losses.combined.CombinedLoss method)

http_max_retries (anemoi.training.schemas.diagnostics.MlflowSchema attribute)

I

identity (anemoi.training.schemas.diagnostics.AzureMlflowSchema attribute) input (anemoi.training.schemas.system.SystemSchema attribute) InputSchema (class in anemoi.training.schemas.system) interval (anemoi.training.schemas.diagnostics.LoggingSchema attribute) (anemoi.training.schemas.diagnostics.WandbSchema attribute) iter_leaf_losses() (anemoi.training.losses.base.BaseLoss method) (anemoi.training.losses.base.BaseLossWrapper method) (anemoi.training.losses.combined.CombinedLoss method)

iter_plot_samples() (anemoi.training.diagnostics.callbacks.plot_adapter.AutoencoderPlotAdapter method) (anemoi.training.diagnostics.callbacks.plot_adapter.BasePlotAdapter method) (anemoi.training.diagnostics.callbacks.plot_adapter.EnsemblePlotAdapterWrapper method) (anemoi.training.diagnostics.callbacks.plot_adapter.ForecasterPlotAdapter method) (anemoi.training.diagnostics.callbacks.plot_adapter.TemporalDownscalerPlotAdapter method)

K

keep_batch_sharded (anemoi.training.train.methods.base.BaseTrainingModule attribute)

L

latlon_bbox (anemoi.training.schemas.diagnostics.FocusAreaSchema attribute) LearningRateMonitor (class in anemoi.training.diagnostics.callbacks.optimiser) limit (anemoi.training.schemas.diagnostics.TimeLimitSchema attribute) load_training_runtime_state_dict() (anemoi.training.tasks.base.BaseTask method) log (anemoi.training.schemas.diagnostics.DiagnosticsSchema attribute) log_extra() (anemoi.training.tasks.base.BaseTask method) log_hyperparams (anemoi.training.schemas.diagnostics.MlflowSchema attribute) log_model (anemoi.training.schemas.diagnostics.MlflowSchema attribute) (anemoi.training.schemas.diagnostics.WandbSchema attribute)

log_optimizer() (anemoi.training.train.methods.base.BaseTrainingModule static method) LoggingSchema (class in anemoi.training.schemas.diagnostics) logs (anemoi.training.schemas.system.OutputSchema attribute) Logs (class in anemoi.training.schemas.system) loss (anemoi.training.train.methods.base.BaseTrainingModule attribute) loss_matrices_path (anemoi.training.schemas.system.InputSchema attribute) LossFactoryContextKey (class in anemoi.training.losses.base) lr_scheduler_step() (anemoi.training.train.methods.base.BaseTrainingModule method)

M

mask_attr_name (anemoi.training.schemas.diagnostics.FocusAreaSchema attribute) mask_nans() (anemoi.training.losses.base.BaseLoss method) MatplotlibColormapClevelsSchema (class in anemoi.training.schemas.diagnostics) MatplotlibColormapSchema (class in anemoi.training.schemas.diagnostics) max_params_length (anemoi.training.schemas.diagnostics.MlflowSchema attribute) members (anemoi.training.schemas.diagnostics.PlotEnsSampleSchema attribute) memory (anemoi.training.schemas.diagnostics.BenchmarkProfilerSchema attribute) MemorySchema (class in anemoi.training.schemas.diagnostics) metadata (anemoi.training.data.multidataset.MultiDataset property) metrics (anemoi.training.train.methods.base.BaseTrainingModule attribute) min_delta (anemoi.training.schemas.diagnostics.EarlyStoppingSchema attribute) mlflow (anemoi.training.schemas.diagnostics.LoggingSchema attribute) (anemoi.training.schemas.system.Logs attribute) MlflowSchema (class in anemoi.training.schemas.diagnostics) mode (anemoi.training.schemas.diagnostics.EarlyStoppingSchema attribute) model (anemoi.training.train.methods.base.BaseTrainingModule attribute) model_config (anemoi.graphs.schemas.base_graph.BaseGraphSchema attribute) (anemoi.graphs.schemas.base_graph.EdgeSchema attribute) (anemoi.graphs.schemas.base_graph.NodeSchema attribute) (anemoi.training.diagnostics.callbacks.plot.PlottingSettings attribute) (anemoi.training.schemas.data.DataSchema attribute) (anemoi.training.schemas.data.DatasetDataSchema attribute) (anemoi.training.schemas.diagnostics.AzureMlflowSchema attribute) (anemoi.training.schemas.diagnostics.BenchmarkProfilerSchema attribute) (anemoi.training.schemas.diagnostics.CheckpointSchema attribute) (anemoi.training.schemas.diagnostics.Debug attribute) (anemoi.training.schemas.diagnostics.DiagnosticsSchema attribute) (anemoi.training.schemas.diagnostics.DistinctipyColormapSchema attribute) (anemoi.training.schemas.diagnostics.EarlyStoppingSchema attribute) (anemoi.training.schemas.diagnostics.FocusAreaSchema attribute) (anemoi.training.schemas.diagnostics.GraphTrainableFeaturesPlotSchema attribute) (anemoi.training.schemas.diagnostics.LoggingSchema attribute) (anemoi.training.schemas.diagnostics.MatplotlibColormapClevelsSchema attribute) (anemoi.training.schemas.diagnostics.MatplotlibColormapSchema attribute) (anemoi.training.schemas.diagnostics.MemorySchema attribute) (anemoi.training.schemas.diagnostics.MlflowSchema attribute) (anemoi.training.schemas.diagnostics.PlotEnsSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotHistogramSchema attribute) (anemoi.training.schemas.diagnostics.PlotLossSchema attribute) (anemoi.training.schemas.diagnostics.PlotSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotSchema attribute) (anemoi.training.schemas.diagnostics.PlotSpectrumSchema attribute) (anemoi.training.schemas.diagnostics.PlottingFrequency attribute) (anemoi.training.schemas.diagnostics.Profiling attribute) (anemoi.training.schemas.diagnostics.ProgressBarSchema attribute) (anemoi.training.schemas.diagnostics.Snapshot attribute) (anemoi.training.schemas.diagnostics.TimeLimitSchema attribute) (anemoi.training.schemas.diagnostics.WandbSchema attribute) (anemoi.training.schemas.system.CheckpointsSchema attribute) (anemoi.training.schemas.system.HardwareSchema attribute) (anemoi.training.schemas.system.InputSchema attribute) (anemoi.training.schemas.system.Logs attribute) (anemoi.training.schemas.system.OutputSchema attribute) (anemoi.training.schemas.system.SystemSchema attribute)

model_summary (anemoi.training.schemas.diagnostics.BenchmarkProfilerSchema attribute) module anemoi.graphs.schemas.base_graph anemoi.training.data.multidataset anemoi.training.diagnostics.callbacks.checkpoint anemoi.training.diagnostics.callbacks.evaluation anemoi.training.diagnostics.callbacks.optimiser anemoi.training.diagnostics.callbacks.plot anemoi.training.diagnostics.callbacks.plot_adapter anemoi.training.diagnostics.callbacks.provenance anemoi.training.losses.base anemoi.training.losses.combined anemoi.training.losses.utils anemoi.training.optimizers.AdEMAMix anemoi.training.schemas.data anemoi.training.schemas.diagnostics anemoi.training.schemas.system anemoi.training.tasks.base anemoi.training.tasks.timeless anemoi.training.train.methods.base anemoi.training.train.methods.edm_diffusion anemoi.training.train.methods.ensemble anemoi.training.train.methods.single anemoi.training.train.methods.stochastic_interpolant anemoi.training.train.methods.transport monitor (anemoi.training.schemas.diagnostics.EarlyStoppingSchema attribute) MultiDataset (class in anemoi.training.data.multidataset)

N

n_colors (anemoi.training.schemas.diagnostics.DistinctipyColormapSchema attribute) n_step_input (anemoi.training.train.methods.base.BaseTrainingModule attribute) n_step_output (anemoi.training.train.methods.base.BaseTrainingModule attribute) name (anemoi.training.losses.base.BaseLoss property) (anemoi.training.schemas.diagnostics.FocusAreaSchema attribute) (anemoi.training.schemas.diagnostics.MatplotlibColormapSchema attribute) (anemoi.training.tasks.base.BaseTask attribute) name_to_index (anemoi.training.data.multidataset.MultiDataset property) needs_shard_layout_info (anemoi.training.losses.base.BaseLoss property) (anemoi.training.losses.base.BaseLossWrapper property) (anemoi.training.losses.combined.CombinedLoss property) node_builder (anemoi.graphs.schemas.base_graph.NodeSchema attribute)

nodes (anemoi.graphs.schemas.base_graph.BaseGraphSchema attribute) NodeSchema (class in anemoi.graphs.schemas.base_graph) num_features (anemoi.training.schemas.data.DataSchema attribute), [1] num_gpus_per_ensemble (anemoi.training.schemas.system.HardwareSchema attribute) num_gpus_per_model (anemoi.training.schemas.system.HardwareSchema attribute) num_gpus_per_node (anemoi.training.schemas.system.HardwareSchema attribute) num_input_timesteps (anemoi.training.tasks.base.BaseTask property) num_inputs (anemoi.training.tasks.base.BaseTask attribute) num_models_saved (anemoi.training.schemas.diagnostics.CheckpointSchema attribute) num_nodes (anemoi.training.schemas.system.HardwareSchema attribute) num_output_timesteps (anemoi.training.tasks.base.BaseTask property) num_outputs (anemoi.training.tasks.base.BaseTask attribute) num_steps (anemoi.training.tasks.base.BaseTask property)

O

offline (anemoi.training.schemas.diagnostics.AzureMlflowSchema attribute) (anemoi.training.schemas.diagnostics.MlflowSchema attribute) (anemoi.training.schemas.diagnostics.WandbSchema attribute) on_after_batch_transfer() (anemoi.training.train.methods.base.BaseTrainingModule method) on_fit_start() (anemoi.training.diagnostics.callbacks.plot.BasePerEpochPlotCallback method) (anemoi.training.diagnostics.callbacks.plot.BasePlotCallback method) on_load_checkpoint() (anemoi.training.diagnostics.callbacks.provenance.ParentUUIDCallback method) (anemoi.training.train.methods.base.BaseTrainingModule method) on_resume_create_child (anemoi.training.schemas.diagnostics.MlflowSchema attribute) on_save_checkpoint() (anemoi.training.train.methods.base.BaseTrainingModule method) on_train_end() (anemoi.training.diagnostics.callbacks.checkpoint.AnemoiCheckpoint method)

on_train_epoch_end() (anemoi.training.tasks.base.BaseTask method) (anemoi.training.train.methods.base.BaseTrainingModule method) on_train_start() (anemoi.training.diagnostics.callbacks.checkpoint.AnemoiCheckpoint method) on_validation_batch_end() (anemoi.training.diagnostics.callbacks.evaluation.RolloutEval method) (anemoi.training.diagnostics.callbacks.plot.BasePerBatchPlotCallback method) (anemoi.training.diagnostics.callbacks.plot.PlotLoss method) on_validation_epoch_end() (anemoi.training.diagnostics.callbacks.plot.BasePerEpochPlotCallback method) output (anemoi.training.schemas.system.SystemSchema attribute) output_mask (anemoi.training.train.methods.base.BaseTrainingModule attribute) OutputSchema (class in anemoi.training.schemas.system) overwrite (anemoi.graphs.schemas.base_graph.BaseGraphSchema attribute)

P

parameter_groups (anemoi.training.schemas.diagnostics.PlotLossSchema attribute) parameters (anemoi.training.schemas.diagnostics.PlotEnsSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotHistogramSchema attribute) (anemoi.training.schemas.diagnostics.PlotSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotSpectrumSchema attribute) (anemoi.training.schemas.diagnostics.WandbSchema attribute) ParentUUIDCallback (class in anemoi.training.diagnostics.callbacks.provenance) patience (anemoi.training.schemas.diagnostics.EarlyStoppingSchema attribute) per_sample (anemoi.training.schemas.diagnostics.PlotEnsSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotSampleSchema attribute) per_worker_init() (anemoi.training.data.multidataset.MultiDataset method) plot (anemoi.training.schemas.diagnostics.DiagnosticsSchema attribute) plot() (anemoi.training.diagnostics.callbacks.plot.BasePlotCallback method) plot_adapter (anemoi.training.train.methods.base.BaseTrainingModule property) (anemoi.training.train.methods.ensemble.EnsembleTraining property) (anemoi.training.train.methods.transport.BaseTransportTraining property) PlotEnsSample (class in anemoi.training.diagnostics.callbacks.plot) PlotEnsSampleSchema (class in anemoi.training.schemas.diagnostics) PlotHistogram (class in anemoi.training.diagnostics.callbacks.plot) PlotHistogramSchema (class in anemoi.training.schemas.diagnostics) PlotLoss (class in anemoi.training.diagnostics.callbacks.plot) PlotLossSchema (class in anemoi.training.schemas.diagnostics) plots (anemoi.training.schemas.system.OutputSchema attribute) PlotSample (class in anemoi.training.diagnostics.callbacks.plot) PlotSampleSchema (class in anemoi.training.schemas.diagnostics) PlotSchema (class in anemoi.training.schemas.diagnostics) PlotSpectrum (class in anemoi.training.diagnostics.callbacks.plot)

PlotSpectrumSchema (class in anemoi.training.schemas.diagnostics) PlottingFrequency (class in anemoi.training.schemas.diagnostics) PlottingSettings (class in anemoi.training.diagnostics.callbacks.plot) precip_and_related_fields (anemoi.training.schemas.diagnostics.PlotEnsSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotHistogramSchema attribute) (anemoi.training.schemas.diagnostics.PlotSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotSchema attribute) prediction_label (anemoi.training.schemas.diagnostics.PlotSampleSchema attribute) prediction_mode (anemoi.training.train.methods.transport.BaseTransportTraining property) PredictionMode (class in anemoi.training.train.methods.transport) prefix (anemoi.training.schemas.diagnostics.MlflowSchema attribute) prepare_loss_batch() (anemoi.training.diagnostics.callbacks.plot_adapter.BasePlotAdapter method) (anemoi.training.diagnostics.callbacks.plot_adapter.EnsemblePlotAdapterWrapper method) prepare_target() (anemoi.training.train.methods.transport.TendencyPredictionMode method) print_memory_summary (anemoi.training.schemas.diagnostics.DiagnosticsSchema attribute) print_variable_scaling() (in module anemoi.training.losses.utils) process() (anemoi.training.diagnostics.callbacks.plot.BasePlotAdditionalMetrics method) process_dataloader() (anemoi.training.distributed.strategy.DDPEnsGroupStrategy method) (anemoi.training.distributed.strategy.DDPGroupStrategy method) process_output_tensor() (anemoi.training.diagnostics.callbacks.plot.BasePlotAdditionalMetrics method) processors (anemoi.training.schemas.data.DatasetDataSchema attribute) profiler (anemoi.training.schemas.system.OutputSchema attribute) Profiling (class in anemoi.training.schemas.diagnostics) progress_bar (anemoi.training.schemas.diagnostics.DiagnosticsSchema attribute) ProgressBarSchema (class in anemoi.training.schemas.diagnostics) project (anemoi.training.schemas.diagnostics.WandbSchema attribute) project_name (anemoi.training.schemas.diagnostics.MlflowSchema attribute) projection_kind (anemoi.training.schemas.diagnostics.PlotSchema attribute)

R

record_file (anemoi.training.schemas.diagnostics.TimeLimitSchema attribute) reduce() (anemoi.training.losses.base.BaseLoss method) reduce_data_output_target_to_model_output() (anemoi.training.train.methods.transport.BaseTransportTraining method) refresh_rate (anemoi.training.schemas.diagnostics.ProgressBarSchema attribute) resolution (anemoi.training.data.multidataset.MultiDataset property) (anemoi.training.schemas.data.DataSchema attribute)

resource_group (anemoi.training.schemas.diagnostics.AzureMlflowSchema attribute) RolloutEval (class in anemoi.training.diagnostics.callbacks.evaluation) root (anemoi.training.schemas.system.CheckpointsSchema attribute) (anemoi.training.schemas.system.OutputSchema attribute) run_name (anemoi.training.schemas.diagnostics.MlflowSchema attribute)

S

sample_idx (anemoi.training.schemas.diagnostics.PlotEnsSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotHistogramSchema attribute) (anemoi.training.schemas.diagnostics.PlotSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotSpectrumSchema attribute) save_dir (anemoi.training.schemas.diagnostics.MlflowSchema attribute) save_frequency (anemoi.training.schemas.diagnostics.CheckpointSchema attribute) scale() (anemoi.training.losses.base.BaseLoss method) scalers (anemoi.training.train.methods.base.BaseTrainingModule attribute) schedule() (anemoi.training.diagnostics.callbacks.plot.AsyncPlotExecutor method) (anemoi.training.diagnostics.callbacks.plot.BasePlotExecutor method) (anemoi.training.diagnostics.callbacks.plot.SyncPlotExecutor method) select_members() (anemoi.training.diagnostics.callbacks.plot_adapter.BasePlotAdapter method) (anemoi.training.diagnostics.callbacks.plot_adapter.EnsemblePlotAdapterWrapper method) set_comm_group_info() (anemoi.training.data.multidataset.MultiDataset method) set_ens_comm_group_info() (anemoi.training.data.multidataset.MultiDataset method) set_epoch() (anemoi.training.data.multidataset.MultiDataset method) setup() (anemoi.training.train.methods.base.BaseTrainingModule method) shard_shapes (anemoi.training.data.multidataset.MultiDataset property) shutdown() (anemoi.training.diagnostics.callbacks.plot.AsyncPlotExecutor method) (anemoi.training.diagnostics.callbacks.plot.BasePlotExecutor method) (anemoi.training.diagnostics.callbacks.plot.SyncPlotExecutor method)

SingleTraining (class in anemoi.training.train.methods.single) snapshot (anemoi.training.schemas.diagnostics.BenchmarkProfilerSchema attribute) Snapshot (class in anemoi.training.schemas.diagnostics) sort_and_color_by_parameter_group() (anemoi.training.diagnostics.callbacks.plot.PlotLoss method) source_name (anemoi.graphs.schemas.base_graph.EdgeSchema attribute) speed (anemoi.training.schemas.diagnostics.BenchmarkProfilerSchema attribute) StatePredictionMode (class in anemoi.training.train.methods.transport) statistics (anemoi.training.data.multidataset.MultiDataset property) step() (anemoi.training.optimizers.AdEMAMix.AdEMAMix method) steps (anemoi.training.schemas.diagnostics.MemorySchema attribute) (anemoi.training.schemas.diagnostics.Snapshot attribute) steps() (anemoi.training.tasks.base.BaseTask method) StochasticInterpolantTransportObjective (class in anemoi.training.train.methods.stochastic_interpolant) stopping_threshold (anemoi.training.schemas.diagnostics.EarlyStoppingSchema attribute) strict (anemoi.training.schemas.diagnostics.EarlyStoppingSchema attribute) subscription_id (anemoi.training.schemas.diagnostics.AzureMlflowSchema attribute) supporting_arrays (anemoi.training.data.multidataset.MultiDataset property) SyncPlotExecutor (class in anemoi.training.diagnostics.callbacks.plot) system (anemoi.training.schemas.diagnostics.BenchmarkProfilerSchema attribute) (anemoi.training.schemas.diagnostics.MlflowSchema attribute) SystemSchema (class in anemoi.training.schemas.system)

T

target (anemoi.training.schemas.data.DatasetDataSchema attribute) target_ (anemoi.training.schemas.diagnostics.DistinctipyColormapSchema attribute) (anemoi.training.schemas.diagnostics.GraphTrainableFeaturesPlotSchema attribute) (anemoi.training.schemas.diagnostics.MatplotlibColormapClevelsSchema attribute) (anemoi.training.schemas.diagnostics.MatplotlibColormapSchema attribute) (anemoi.training.schemas.diagnostics.PlotEnsSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotHistogramSchema attribute) (anemoi.training.schemas.diagnostics.PlotLossSchema attribute) (anemoi.training.schemas.diagnostics.PlotSampleSchema attribute) (anemoi.training.schemas.diagnostics.PlotSpectrumSchema attribute) (anemoi.training.schemas.diagnostics.ProgressBarSchema attribute) (anemoi.training.schemas.diagnostics.TimeLimitSchema attribute) target_name (anemoi.graphs.schemas.base_graph.EdgeSchema attribute) teardown() (anemoi.training.diagnostics.callbacks.plot.BasePlotCallback method) TemporalDownscalerPlotAdapter (class in anemoi.training.diagnostics.callbacks.plot_adapter)

TendencyPredictionMode (class in anemoi.training.train.methods.transport) tensorboard (anemoi.training.schemas.system.Logs attribute) terminal (anemoi.training.schemas.diagnostics.AzureMlflowSchema attribute) (anemoi.training.schemas.diagnostics.MlflowSchema attribute) time (anemoi.training.schemas.diagnostics.BenchmarkProfilerSchema attribute) TimeLimitSchema (class in anemoi.training.schemas.diagnostics) timestep (anemoi.training.schemas.data.DataSchema attribute) trace_rank0_only (anemoi.training.schemas.diagnostics.MemorySchema attribute) tracking_uri (anemoi.training.schemas.diagnostics.MlflowSchema attribute) training_runtime_state_dict() (anemoi.training.tasks.base.BaseTask method) training_step() (anemoi.training.train.methods.base.BaseTrainingModule method) transfer_batch_to_device() (anemoi.training.train.methods.base.BaseTrainingModule method) transport_objective (anemoi.training.train.methods.transport.TransportTraining property) TransportTraining (class in anemoi.training.train.methods.transport) truncation (anemoi.training.schemas.system.InputSchema attribute) truncation_inv (anemoi.training.schemas.system.InputSchema attribute)

U

update_scaler() (anemoi.training.losses.base.BaseLoss method) (anemoi.training.losses.base.BaseLossWrapper method) (anemoi.training.losses.combined.CombinedLoss method)

update_scalers() (anemoi.training.train.methods.base.BaseTrainingModule method)

V

val_metric_ranges (anemoi.training.train.methods.base.BaseTrainingModule attribute) validation_step() (anemoi.training.train.methods.base.BaseTrainingModule method) variables (anemoi.training.data.multidataset.MultiDataset property) (anemoi.training.schemas.diagnostics.DistinctipyColormapSchema attribute) (anemoi.training.schemas.diagnostics.MatplotlibColormapClevelsSchema attribute) (anemoi.training.schemas.diagnostics.MatplotlibColormapSchema attribute)

verbose (anemoi.training.schemas.diagnostics.EarlyStoppingSchema attribute) (anemoi.training.schemas.diagnostics.Profiling attribute)

W

wandb (anemoi.training.schemas.diagnostics.LoggingSchema attribute) (anemoi.training.schemas.system.Logs attribute) WandbSchema (class in anemoi.training.schemas.diagnostics)

warm_start (anemoi.training.schemas.system.InputSchema attribute) warmup (anemoi.training.schemas.diagnostics.MemorySchema attribute) (anemoi.training.schemas.diagnostics.Snapshot attribute) workspace_name (anemoi.training.schemas.diagnostics.AzureMlflowSchema attribute)