Index

A

affine_transform() (in module anemoi.models.preprocessing.mappings) aggregate() (anemoi.models.layers.conv.GraphConv method) all_to_all_transpose() (in module anemoi.models.distributed.graph) anemoi.models.data_indices.collection module anemoi.models.data_indices.index module anemoi.models.data_indices.tensor module anemoi.models.distributed.graph module anemoi.models.distributed.khop_edges module anemoi.models.distributed.shapes module anemoi.models.distributed.utils module anemoi.models.interface module anemoi.models.layers.activations module anemoi.models.layers.attention module anemoi.models.layers.block module anemoi.models.layers.conv module anemoi.models.layers.graph module anemoi.models.layers.graph_provider module anemoi.models.layers.mapper module anemoi.models.layers.mlp module anemoi.models.layers.normalization module

anemoi.models.layers.processor module anemoi.models.layers.utils module anemoi.models.migrations.migrator module anemoi.models.migrations.setup_context module anemoi.models.preprocessing module anemoi.models.preprocessing.imputer module anemoi.models.preprocessing.mappings module anemoi.models.preprocessing.normalizer module anemoi.models.preprocessing.remapper module anemoi.models.schemas.decoder module anemoi.models.schemas.encoder module anemoi.models.schemas.models module anemoi.models.schemas.processor module AnemoiEnsModelEncProcDec (class in anemoi.models.models.ens_encoder_processor_decoder) AnemoiModelEncProcDec (class in anemoi.models.models.encoder_processor_decoder) AnemoiModelEncProcDecHierarchical (class in anemoi.models.models.hierarchical) AnemoiModelInterface (class in anemoi.models.interface) asinh_converter() (in module anemoi.models.preprocessing.mappings) atanh_converter() (in module anemoi.models.preprocessing.mappings) attention_implementation (anemoi.models.schemas.decoder.TransformerDecoderSchema attribute) (anemoi.models.schemas.encoder.TransformerEncoderSchema attribute) (anemoi.models.schemas.processor.TransformerProcessorSchema attribute) attr_ndims (anemoi.models.layers.graph.NamedNodesAttributes attribute) autocast (anemoi.models.schemas.models.NoiseConditioningSchema attribute) AutocastLayerNorm (class in anemoi.models.layers.normalization)

B

BaseBlock (class in anemoi.models.layers.block) BaseGraphProvider (class in anemoi.models.layers.graph_provider) BaseImputer (class in anemoi.models.preprocessing.imputer) BaseIndex (class in anemoi.models.data_indices.index) BaseMapper (class in anemoi.models.layers.mapper) BaseModelSchema (class in anemoi.models.schemas.models) BasePreprocessor (class in anemoi.models.preprocessing) BaseProcessor (class in anemoi.models.layers.processor) BaseTensorIndex (class in anemoi.models.data_indices.tensor)

BipartiteGraphShardInfo (class in anemoi.models.distributed.shapes) Boolean1DSchema (class in anemoi.models.schemas.models) bounding (anemoi.models.schemas.models.BaseModelSchema attribute) boxcox_converter() (in module anemoi.models.preprocessing.mappings) build_feedforward_layer() (in module anemoi.models.layers.mlp) build_feedforward_modules() (in module anemoi.models.layers.mlp) build_graph() (anemoi.models.layers.graph_provider.DynamicGraphProvider method) build_graph_partition() (in module anemoi.models.distributed.khop_edges) build_graph_partition_from_shard_info() (in module anemoi.models.distributed.khop_edges) build_layers() (anemoi.models.layers.processor.BaseProcessor method)

C

CheckpointWrapper (class in anemoi.models.layers.utils) cls (in module anemoi.models.data_indices.collection) compare_variables() (anemoi.models.data_indices.collection.IndexCollection method) compile (anemoi.models.schemas.models.BaseModelSchema attribute) compute_mlp_hidden_dim() (in module anemoi.models.layers.utils) condition_on_residual (anemoi.models.schemas.models.EnsModelSchema attribute) (anemoi.models.schemas.models.TransportTendModelSchema attribute)

ConditionalLayerNorm (class in anemoi.models.layers.normalization) config (anemoi.models.interface.AnemoiModelInterface attribute) ConstantImputer (class in anemoi.models.preprocessing.imputer) convert_ (anemoi.models.schemas.models.Model attribute) CopyImputer (class in anemoi.models.preprocessing.imputer) create_graph_provider() (in module anemoi.models.layers.graph_provider) create_sliding_window_mask() (anemoi.models.layers.attention.SDPAAttentionWrapper method) CustomRelu (class in anemoi.models.layers.activations)

D

data (anemoi.models.schemas.models.TrainableParameters attribute) data_indices (anemoi.models.interface.AnemoiModelInterface attribute) DataIndex (class in anemoi.models.data_indices.index) define_fixed_attributes() (anemoi.models.layers.graph.NamedNodesAttributes method) DefinedModels (class in anemoi.models.schemas.models) delete_attribute() (anemoi.models.migrations.setup_context.MigrationContext method) delete_module() (anemoi.models.migrations.setup_context.MigrationContext method)

displace_boundary_atoms() (in module anemoi.models.preprocessing.mappings) drop_unconnected_src_nodes() (in module anemoi.models.distributed.khop_edges) dropout_p (anemoi.models.schemas.decoder.TransformerDecoderSchema attribute) (anemoi.models.schemas.encoder.TransformerEncoderSchema attribute) (anemoi.models.schemas.processor.PointWiseMLPProcessorSchema attribute) (anemoi.models.schemas.processor.TransformerProcessorSchema attribute) DynamicGraphProvider (class in anemoi.models.layers.graph_provider)

E

edge_dim (anemoi.models.layers.graph_provider.BaseGraphProvider property) (anemoi.models.layers.graph_provider.DynamicGraphProvider property) (anemoi.models.layers.graph_provider.NoOpGraphProvider property) (anemoi.models.layers.graph_provider.ProjectionGraphProvider property) (anemoi.models.layers.graph_provider.StaticGraphProvider property) edge_weight_attribute (anemoi.models.schemas.models.NoiseConditioningSchema attribute)

enable_hierarchical_level_processing (anemoi.models.schemas.models.HierarchicalModelSchema attribute) encoder (anemoi.models.schemas.models.BaseModelSchema attribute) EnsModelSchema (class in anemoi.models.schemas.models) ensure_edges_are_dst_sorted() (in module anemoi.models.distributed.khop_edges) ensure_sharded() (in module anemoi.models.distributed.graph) expand_shard_sizes_to_shapes() (in module anemoi.models.distributed.shapes) expm1_converter() (in module anemoi.models.preprocessing.mappings)

F

fill_metadata() (anemoi.models.models.encoder_processor_decoder.AnemoiModelEncProcDec method) fill_with_value() (anemoi.models.preprocessing.imputer.BaseImputer method) (anemoi.models.preprocessing.imputer.CopyImputer method) final (anemoi.models.migrations.migrator.MigrationMetadata attribute) FlashAttentionWrapper (class in anemoi.models.layers.attention) forward() (anemoi.models.layers.activations.CustomRelu method) (anemoi.models.layers.activations.Sine method) (anemoi.models.layers.attention.FlashAttentionWrapper method) (anemoi.models.layers.attention.MultiHeadCrossAttention method) (anemoi.models.layers.attention.MultiHeadSelfAttention method) (anemoi.models.layers.attention.SDPAAttentionWrapper method) (anemoi.models.layers.block.BaseBlock method) (anemoi.models.layers.block.GraphConvBaseBlock method) (anemoi.models.layers.block.GraphConvMapperBlock method) (anemoi.models.layers.block.GraphConvProcessorBlock method) (anemoi.models.layers.block.GraphTransformerBaseBlock method) (anemoi.models.layers.block.GraphTransformerMapperBlock method) (anemoi.models.layers.block.GraphTransformerProcessorBlock method) (anemoi.models.layers.block.PointWiseMLPProcessorBlock method) (anemoi.models.layers.block.TransformerMapperBlock method) (anemoi.models.layers.block.TransformerProcessorBlock method) (anemoi.models.layers.conv.GraphConv method) (anemoi.models.layers.conv.GraphTransformerConv method) (anemoi.models.layers.graph.NamedNodesAttributes method), [1] (anemoi.models.layers.graph.TrainableTensor method) (anemoi.models.layers.mapper.BaseMapper method) (anemoi.models.layers.mapper.GNNBackwardMapper method) (anemoi.models.layers.mapper.GNNBaseMapper method) (anemoi.models.layers.mapper.GraphTransformerBaseMapper method) (anemoi.models.layers.mapper.GraphTransformerForwardMapper method) (anemoi.models.layers.mapper.PointWiseForwardMapper method) (anemoi.models.layers.mapper.PointWiseMapper method) (anemoi.models.layers.mapper.TransformerBaseMapper method) (anemoi.models.layers.mapper.TransformerForwardMapper method) (anemoi.models.layers.mlp.GatedMLPLayer method) (anemoi.models.layers.mlp.MLP method) (anemoi.models.layers.normalization.AutocastLayerNorm method) (anemoi.models.layers.normalization.ConditionalLayerNorm method) (anemoi.models.layers.processor.BaseProcessor method) (anemoi.models.layers.processor.GNNProcessor method) (anemoi.models.layers.processor.GraphTransformerProcessor method) (anemoi.models.layers.processor.NoOpProcessor method) (anemoi.models.layers.processor.PointWiseMLPProcessor method) (anemoi.models.layers.processor.TransformerProcessor method) (anemoi.models.layers.residual.ScalarOrnsteinConnection method) (anemoi.models.layers.residual.SkipConnection method) (anemoi.models.layers.residual.SpectralOrnsteinConnection method) (anemoi.models.layers.residual.TruncatedConnection method) (anemoi.models.layers.utils.CheckpointWrapper method) (anemoi.models.models.encoder_processor_decoder.AnemoiModelEncProcDec method) (anemoi.models.models.ens_encoder_processor_decoder.AnemoiEnsModelEncProcDec method) (anemoi.models.models.hierarchical.AnemoiModelEncProcDecHierarchical method) (anemoi.models.preprocessing.BasePreprocessor method) (anemoi.models.preprocessing.Processors method)

FractionBoundingSchema (class in anemoi.models.schemas.models) from_config() (anemoi.models.layers.graph_provider.ProjectionGraphProvider class method) from_serialized() (anemoi.models.migrations.migrator.Migration class method)

G

GatedMLPLayer (class in anemoi.models.layers.mlp) gather_tensor() (in module anemoi.models.distributed.graph) get_alibi_slopes() (in module anemoi.models.layers.attention) get_coordinates() (anemoi.models.layers.graph.NamedNodesAttributes method), [1] get_edges() (anemoi.models.layers.graph_provider.BaseGraphProvider method) (anemoi.models.layers.graph_provider.DynamicGraphProvider method) (anemoi.models.layers.graph_provider.NoOpGraphProvider method) (anemoi.models.layers.graph_provider.ProjectionGraphProvider method) (anemoi.models.layers.graph_provider.StaticGraphProvider method) get_memory_format() (in module anemoi.models.distributed.utils) get_nans() (anemoi.models.preprocessing.imputer.BaseImputer method) get_shard_sizes() (in module anemoi.models.distributed.shapes) GNNBackwardMapper (class in anemoi.models.layers.mapper) GNNBaseMapper (class in anemoi.models.layers.mapper) GNNDecoderSchema (class in anemoi.models.schemas.decoder) GNNEncoderSchema (class in anemoi.models.schemas.encoder) GNNForwardMapper (class in anemoi.models.layers.mapper) GNNProcessor (class in anemoi.models.layers.processor)

GNNProcessorSchema (class in anemoi.models.schemas.processor) graph_data (anemoi.models.interface.AnemoiModelInterface attribute) GraphConv (class in anemoi.models.layers.conv) GraphConvBaseBlock (class in anemoi.models.layers.block) GraphConvMapperBlock (class in anemoi.models.layers.block) GraphConvProcessorBlock (class in anemoi.models.layers.block) GraphPartition (class in anemoi.models.distributed.khop_edges) GraphShardInfo (class in anemoi.models.distributed.shapes) GraphTransformerBackwardMapper (class in anemoi.models.layers.mapper) GraphTransformerBaseBlock (class in anemoi.models.layers.block) GraphTransformerBaseMapper (class in anemoi.models.layers.mapper) GraphTransformerConv (class in anemoi.models.layers.conv) GraphTransformerDecoderSchema (class in anemoi.models.schemas.decoder) GraphTransformerEncoderSchema (class in anemoi.models.schemas.encoder) GraphTransformerForwardMapper (class in anemoi.models.layers.mapper) GraphTransformerMapperBlock (class in anemoi.models.layers.block) GraphTransformerProcessor (class in anemoi.models.layers.processor) GraphTransformerProcessorBlock (class in anemoi.models.layers.block) GraphTransformerProcessorSchema (class in anemoi.models.schemas.processor)

H

HardtanhBoundingSchema (class in anemoi.models.schemas.models) hidden (anemoi.models.schemas.models.TrainableParameters attribute)

hidden_nodes_name (anemoi.models.schemas.models.Model attribute) HierarchicalModelSchema (class in anemoi.models.schemas.models)

I

id (anemoi.models.interface.AnemoiModelInterface attribute) IncompatibleCheckpointException IncompleteMigrationScript IndexCollection (class in anemoi.models.data_indices.collection) inference_defaults (anemoi.models.schemas.models.TransportConfig attribute) initialise_data_extractor_zero (anemoi.models.schemas.decoder.GraphTransformerDecoderSchema attribute) (anemoi.models.schemas.decoder.PointWiseBackwardMapperSchema attribute) InputImputer (class in anemoi.models.preprocessing.imputer) InputNormalizer (class in anemoi.models.preprocessing.normalizer) InputTensorIndex (class in anemoi.models.data_indices.tensor) inverse_asinh_converter() (in module anemoi.models.preprocessing.mappings)

inverse_boxcox_converter() (in module anemoi.models.preprocessing.mappings) inverse_displace_boundary_atoms() (in module anemoi.models.preprocessing.mappings) inverse_power_transform() (in module anemoi.models.preprocessing.mappings) inverse_sqrt_converter() (in module anemoi.models.preprocessing.mappings) inverse_transform() (anemoi.models.preprocessing.BasePreprocessor method) (anemoi.models.preprocessing.imputer.BaseImputer method) (anemoi.models.preprocessing.normalizer.InputNormalizer method) (anemoi.models.preprocessing.remapper.Remapper method) is_edge_index_dst_sorted() (in module anemoi.models.distributed.khop_edges) is_sparse (anemoi.models.layers.graph_provider.BaseGraphProvider property) (anemoi.models.layers.graph_provider.ProjectionGraphProvider property)

K

keep_batch_sharded (anemoi.models.schemas.models.BaseModelSchema attribute)

kind (anemoi.models.schemas.models.TransportSourceConfig attribute)

L

latent_skip (anemoi.models.schemas.models.BaseModelSchema attribute) (anemoi.models.schemas.models.Model attribute) layer_kernels (anemoi.models.schemas.models.NoiseConditioningSchema attribute) (anemoi.models.schemas.models.NoiseInjectorSchema attribute) leaky_hardtanh() (in module anemoi.models.layers.activations)

LeakyFractionBoundingSchema (class in anemoi.models.schemas.models) LeakyHardtanhBoundingSchema (class in anemoi.models.schemas.models) LeakyReluBoundingSchema (class in anemoi.models.schemas.models) level_process_num_layers (anemoi.models.schemas.models.HierarchicalModelSchema attribute) load_layer_kernels() (in module anemoi.models.layers.utils) log1p_converter() (in module anemoi.models.preprocessing.mappings)

M

materialise() (anemoi.models.distributed.khop_edges.GraphPartition method) max_val (anemoi.models.schemas.models.FractionBoundingSchema attribute) (anemoi.models.schemas.models.HardtanhBoundingSchema attribute) maybe_checkpoint() (in module anemoi.models.layers.utils) message() (anemoi.models.layers.conv.GraphConv method) (anemoi.models.layers.conv.GraphTransformerConv method) metadata (anemoi.models.interface.AnemoiModelInterface attribute) (anemoi.models.migrations.migrator.Migration attribute) migrate (anemoi.models.migrations.migrator.Migration attribute) migrate_setup (anemoi.models.migrations.migrator.Migration attribute) Migration (class in anemoi.models.migrations.migrator) MigrationContext (class in anemoi.models.migrations.setup_context) MigrationMetadata (class in anemoi.models.migrations.migrator) MigrationOp (class in anemoi.models.migrations.migrator) MigrationVersions (class in anemoi.models.migrations.migrator) min_val (anemoi.models.schemas.models.FractionBoundingSchema attribute) (anemoi.models.schemas.models.HardtanhBoundingSchema attribute) MissingAttribute (class in anemoi.models.migrations.migrator) MLP (class in anemoi.models.layers.mlp) mlp_hidden_ratio (anemoi.models.schemas.processor.PointWiseMLPProcessorSchema attribute) model (anemoi.models.interface.AnemoiModelInterface attribute) (anemoi.models.schemas.models.BaseModelSchema attribute) (anemoi.models.schemas.models.TransportModelSchema attribute) Model (class in anemoi.models.schemas.models) model_config (anemoi.models.schemas.decoder.GNNDecoderSchema attribute) (anemoi.models.schemas.decoder.GraphTransformerDecoderSchema attribute) (anemoi.models.schemas.decoder.PointWiseBackwardMapperSchema attribute) (anemoi.models.schemas.decoder.TransformerDecoderSchema attribute) (anemoi.models.schemas.encoder.GNNEncoderSchema attribute) (anemoi.models.schemas.encoder.GraphTransformerEncoderSchema attribute) (anemoi.models.schemas.encoder.PointWiseForwardMapperSchema attribute) (anemoi.models.schemas.encoder.TransformerEncoderSchema attribute) (anemoi.models.schemas.models.BaseModelSchema attribute) (anemoi.models.schemas.models.Boolean1DSchema attribute) (anemoi.models.schemas.models.EnsModelSchema attribute) (anemoi.models.schemas.models.FractionBoundingSchema attribute) (anemoi.models.schemas.models.HardtanhBoundingSchema attribute) (anemoi.models.schemas.models.HierarchicalModelSchema attribute) (anemoi.models.schemas.models.LeakyFractionBoundingSchema attribute) (anemoi.models.schemas.models.LeakyHardtanhBoundingSchema attribute) (anemoi.models.schemas.models.LeakyReluBoundingSchema attribute) (anemoi.models.schemas.models.Model attribute) (anemoi.models.schemas.models.NoiseConditioningSchema attribute) (anemoi.models.schemas.models.NoiseInjectorSchema attribute) (anemoi.models.schemas.models.NoOpNoiseInjectorSchema attribute) (anemoi.models.schemas.models.NoOutputMaskSchema attribute) (anemoi.models.schemas.models.NormalizedLeakyReluBoundingSchema attribute) (anemoi.models.schemas.models.NormalizedReluBoundingSchema attribute) (anemoi.models.schemas.models.ReluBoundingSchema attribute) (anemoi.models.schemas.models.TrainableParameters attribute) (anemoi.models.schemas.models.TransportConfig attribute) (anemoi.models.schemas.models.TransportModel attribute) (anemoi.models.schemas.models.TransportModelSchema attribute) (anemoi.models.schemas.models.TransportSourceConfig attribute) (anemoi.models.schemas.models.TransportTendModelSchema attribute) (anemoi.models.schemas.processor.GNNProcessorSchema attribute) (anemoi.models.schemas.processor.GraphTransformerProcessorSchema attribute) (anemoi.models.schemas.processor.NoOpProcessorSchema attribute) (anemoi.models.schemas.processor.PointWiseMLPProcessorSchema attribute) (anemoi.models.schemas.processor.TransformerProcessorSchema attribute)

model_is_distributed() (in module anemoi.models.distributed.utils) ModelIndex (class in anemoi.models.data_indices.index) module anemoi.models.data_indices.collection anemoi.models.data_indices.index anemoi.models.data_indices.tensor anemoi.models.distributed.graph anemoi.models.distributed.khop_edges anemoi.models.distributed.shapes anemoi.models.distributed.utils anemoi.models.interface anemoi.models.layers.activations anemoi.models.layers.attention anemoi.models.layers.block anemoi.models.layers.conv anemoi.models.layers.graph anemoi.models.layers.graph_provider anemoi.models.layers.mapper anemoi.models.layers.mlp anemoi.models.layers.normalization anemoi.models.layers.processor anemoi.models.layers.utils anemoi.models.migrations.migrator anemoi.models.migrations.setup_context anemoi.models.preprocessing anemoi.models.preprocessing.imputer anemoi.models.preprocessing.mappings anemoi.models.preprocessing.normalizer anemoi.models.preprocessing.remapper anemoi.models.schemas.decoder anemoi.models.schemas.encoder anemoi.models.schemas.models anemoi.models.schemas.processor move_attribute() (anemoi.models.migrations.setup_context.MigrationContext method) move_module() (anemoi.models.migrations.setup_context.MigrationContext method) MultiHeadCrossAttention (class in anemoi.models.layers.attention) MultiHeadSelfAttention (class in anemoi.models.layers.attention)

N

n_step_input (anemoi.models.interface.AnemoiModelInterface attribute) name (anemoi.models.migrations.migrator.Migration attribute) (anemoi.models.migrations.migrator.SerializedMigration attribute) NamedNodesAttributes (class in anemoi.models.layers.graph) noise_channels (anemoi.models.schemas.models.TransportConfig attribute) noise_channels_dim (anemoi.models.schemas.models.NoiseConditioningSchema attribute) (anemoi.models.schemas.models.NoiseInjectorSchema attribute) noise_cond_dim (anemoi.models.schemas.models.TransportConfig attribute) noise_edges_name (anemoi.models.schemas.models.NoiseConditioningSchema attribute) noise_embedder (anemoi.models.schemas.models.TransportConfig attribute) noise_injector (anemoi.models.schemas.models.EnsModelSchema attribute) noise_matrix (anemoi.models.schemas.models.NoiseConditioningSchema attribute) noise_mlp_hidden_dim (anemoi.models.schemas.models.NoiseConditioningSchema attribute) (anemoi.models.schemas.models.NoiseInjectorSchema attribute) noise_scale (anemoi.models.schemas.models.TransportSourceConfig attribute) noise_std (anemoi.models.schemas.models.NoiseConditioningSchema attribute) (anemoi.models.schemas.models.NoiseInjectorSchema attribute) NoiseConditioningSchema (class in anemoi.models.schemas.models)

NoiseInjectorSchema (class in anemoi.models.schemas.models) noop() (in module anemoi.models.preprocessing.mappings) NoOpGraphProvider (class in anemoi.models.layers.graph_provider) NoOpNoiseInjectorSchema (class in anemoi.models.schemas.models) NoOpProcessor (class in anemoi.models.layers.processor) NoOpProcessorSchema (class in anemoi.models.schemas.processor) NoOutputMaskSchema (class in anemoi.models.schemas.models) normalize_projection_edges_name() (in module anemoi.models.layers.graph_provider) NormalizedLeakyReluBoundingSchema (class in anemoi.models.schemas.models) NormalizedReluBoundingSchema (class in anemoi.models.schemas.models) num_channels (anemoi.models.schemas.models.BaseModelSchema attribute) num_chunks (anemoi.models.schemas.processor.GNNProcessorSchema attribute) (anemoi.models.schemas.processor.GraphTransformerProcessorSchema attribute) (anemoi.models.schemas.processor.TransformerProcessorSchema attribute) num_layers (anemoi.models.schemas.processor.GNNProcessorSchema attribute) (anemoi.models.schemas.processor.GraphTransformerProcessorSchema attribute) (anemoi.models.schemas.processor.PointWiseMLPProcessorSchema attribute) (anemoi.models.schemas.processor.TransformerProcessorSchema attribute) num_nodes (anemoi.models.layers.graph.NamedNodesAttributes attribute)

O

objective (anemoi.models.schemas.models.TransportConfig attribute)

output_mask (anemoi.models.schemas.models.BaseModelSchema attribute) OutputTensorIndex (class in anemoi.models.data_indices.tensor)

P

PointWiseBackwardMapper (class in anemoi.models.layers.mapper) PointWiseBackwardMapperSchema (class in anemoi.models.schemas.decoder) PointWiseForwardMapper (class in anemoi.models.layers.mapper) PointWiseForwardMapperSchema (class in anemoi.models.schemas.encoder) PointWiseMapper (class in anemoi.models.layers.mapper) PointWiseMLPProcessor (class in anemoi.models.layers.processor) PointWiseMLPProcessorBlock (class in anemoi.models.layers.block) PointWiseMLPProcessorSchema (class in anemoi.models.schemas.processor)

positions_for_names() (anemoi.models.data_indices.tensor.BaseTensorIndex method) post_processors (anemoi.models.interface.AnemoiModelInterface attribute) power_transform() (in module anemoi.models.preprocessing.mappings) pre_processors (anemoi.models.interface.AnemoiModelInterface attribute) predict_step() (anemoi.models.interface.AnemoiModelInterface method) processor (anemoi.models.schemas.models.BaseModelSchema attribute) Processors (class in anemoi.models.preprocessing) ProjectionGraphProvider (class in anemoi.models.layers.graph_provider)

Q

qk_norm (anemoi.models.schemas.decoder.GraphTransformerDecoderSchema attribute) (anemoi.models.schemas.encoder.GraphTransformerEncoderSchema attribute) (anemoi.models.schemas.processor.GraphTransformerProcessorSchema attribute) (anemoi.models.schemas.processor.TransformerProcessorSchema attribute)

R

reduce_shard_tensor() (in module anemoi.models.distributed.graph) reduce_tensor() (in module anemoi.models.distributed.graph) register_coordinates() (anemoi.models.layers.graph.NamedNodesAttributes method) register_tensor() (anemoi.models.layers.graph.NamedNodesAttributes method) ReluBoundingSchema (class in anemoi.models.schemas.models)

Remapper (class in anemoi.models.preprocessing.remapper) residual (anemoi.models.schemas.models.BaseModelSchema attribute) rho (anemoi.models.schemas.models.TransportConfig attribute) row_normalize_noise_matrix (anemoi.models.schemas.models.NoiseConditioningSchema attribute) run_layers() (anemoi.models.layers.processor.BaseProcessor method)

S

SaveCkpt (class in anemoi.models.migrations.migrator) ScalarOrnsteinConnection (class in anemoi.models.layers.residual) scale (anemoi.models.schemas.models.TransportSourceConfig attribute) SDPAAttentionWrapper (class in anemoi.models.layers.attention) serialize() (anemoi.models.migrations.migrator.Migration method) SerializedMigration (class in anemoi.models.migrations.migrator) shard_edges_1hop() (in module anemoi.models.distributed.khop_edges) shard_graph_to_local() (in module anemoi.models.distributed.khop_edges) shard_output_seq() (anemoi.models.layers.block.GraphTransformerBaseBlock method) shard_qkve_heads() (anemoi.models.layers.block.GraphTransformerBaseBlock method) shard_tensor() (in module anemoi.models.distributed.graph) si_alpha_schedule (anemoi.models.schemas.models.TransportConfig attribute) si_beta_schedule (anemoi.models.schemas.models.TransportConfig attribute) si_noise_scale (anemoi.models.schemas.models.TransportConfig attribute) si_sigma_schedule (anemoi.models.schemas.models.TransportConfig attribute) sigma_data (anemoi.models.schemas.models.TransportConfig attribute) sigma_max (anemoi.models.schemas.models.TransportConfig attribute) sigma_min (anemoi.models.schemas.models.TransportConfig attribute) signature (anemoi.models.migrations.migrator.Migration attribute) (anemoi.models.migrations.migrator.SerializedMigration attribute)

Sine (class in anemoi.models.layers.activations) SkipConnection (class in anemoi.models.layers.residual) softcap (anemoi.models.schemas.decoder.TransformerDecoderSchema attribute) (anemoi.models.schemas.encoder.TransformerEncoderSchema attribute) (anemoi.models.schemas.processor.TransformerProcessorSchema attribute) sort_edge_index_by_dst() (in module anemoi.models.distributed.khop_edges) sort_edges_1hop_chunks() (in module anemoi.models.distributed.khop_edges) source (anemoi.models.schemas.models.TransportConfig attribute) SpectralOrnsteinConnection (class in anemoi.models.layers.residual) sqrt_converter() (in module anemoi.models.preprocessing.mappings) StaticGraphProvider (class in anemoi.models.layers.graph_provider) statistics (anemoi.models.interface.AnemoiModelInterface attribute) statistics_tendencies (anemoi.models.interface.AnemoiModelInterface attribute) StepwiseProcessors (class in anemoi.models.preprocessing) sub_graph_edge_attributes (anemoi.models.schemas.decoder.GraphTransformerDecoderSchema attribute) (anemoi.models.schemas.encoder.GraphTransformerEncoderSchema attribute) (anemoi.models.schemas.processor.GraphTransformerProcessorSchema attribute) supporting_arrays (anemoi.models.interface.AnemoiModelInterface attribute) sync_tensor() (in module anemoi.models.distributed.graph)

T

target_ (anemoi.models.schemas.decoder.GNNDecoderSchema attribute) (anemoi.models.schemas.decoder.GraphTransformerDecoderSchema attribute) (anemoi.models.schemas.decoder.PointWiseBackwardMapperSchema attribute) (anemoi.models.schemas.decoder.TransformerDecoderSchema attribute) (anemoi.models.schemas.encoder.GNNEncoderSchema attribute) (anemoi.models.schemas.encoder.GraphTransformerEncoderSchema attribute) (anemoi.models.schemas.encoder.PointWiseForwardMapperSchema attribute) (anemoi.models.schemas.encoder.TransformerEncoderSchema attribute) (anemoi.models.schemas.models.FractionBoundingSchema attribute) (anemoi.models.schemas.models.HardtanhBoundingSchema attribute) (anemoi.models.schemas.models.LeakyFractionBoundingSchema attribute) (anemoi.models.schemas.models.LeakyHardtanhBoundingSchema attribute) (anemoi.models.schemas.models.LeakyReluBoundingSchema attribute) (anemoi.models.schemas.models.Model attribute) (anemoi.models.schemas.models.NoiseConditioningSchema attribute) (anemoi.models.schemas.models.NoiseInjectorSchema attribute) (anemoi.models.schemas.models.NoOpNoiseInjectorSchema attribute) (anemoi.models.schemas.models.NormalizedLeakyReluBoundingSchema attribute) (anemoi.models.schemas.models.ReluBoundingSchema attribute) (anemoi.models.schemas.processor.GNNProcessorSchema attribute) (anemoi.models.schemas.processor.GraphTransformerProcessorSchema attribute) (anemoi.models.schemas.processor.NoOpProcessorSchema attribute) (anemoi.models.schemas.processor.PointWiseMLPProcessorSchema attribute) (anemoi.models.schemas.processor.TransformerProcessorSchema attribute) total_var (anemoi.models.schemas.models.FractionBoundingSchema attribute) trainable_parameters (anemoi.models.schemas.models.BaseModelSchema attribute)

trainable_size (anemoi.models.schemas.decoder.GraphTransformerDecoderSchema attribute) (anemoi.models.schemas.encoder.GraphTransformerEncoderSchema attribute) (anemoi.models.schemas.processor.GraphTransformerProcessorSchema attribute) trainable_tensors (anemoi.models.layers.graph.NamedNodesAttributes attribute) TrainableParameters (class in anemoi.models.schemas.models) TrainableTensor (class in anemoi.models.layers.graph) training_condition (anemoi.models.schemas.models.TransportConfig attribute) transform() (anemoi.models.preprocessing.BasePreprocessor method) (anemoi.models.preprocessing.imputer.BaseImputer method) (anemoi.models.preprocessing.normalizer.InputNormalizer method) (anemoi.models.preprocessing.remapper.Remapper method) TransformerBackwardMapper (class in anemoi.models.layers.mapper) TransformerBaseMapper (class in anemoi.models.layers.mapper) TransformerDecoderSchema (class in anemoi.models.schemas.decoder) TransformerEncoderSchema (class in anemoi.models.schemas.encoder) TransformerForwardMapper (class in anemoi.models.layers.mapper) TransformerMapperBlock (class in anemoi.models.layers.block) TransformerProcessor (class in anemoi.models.layers.processor) TransformerProcessorBlock (class in anemoi.models.layers.block) TransformerProcessorSchema (class in anemoi.models.schemas.processor) transport (anemoi.models.schemas.models.TransportModel attribute) TransportConfig (class in anemoi.models.schemas.models) TransportModel (class in anemoi.models.schemas.models) TransportModelSchema (class in anemoi.models.schemas.models) TransportSourceConfig (class in anemoi.models.schemas.models) TransportTendModelSchema (class in anemoi.models.schemas.models) TruncatedConnection (class in anemoi.models.layers.residual)

U

use_alibi_slopes (anemoi.models.schemas.decoder.TransformerDecoderSchema attribute) (anemoi.models.schemas.encoder.TransformerEncoderSchema attribute) (anemoi.models.schemas.processor.TransformerProcessorSchema attribute)

use_rotary_embeddings (anemoi.models.schemas.decoder.TransformerDecoderSchema attribute)

V

variables (anemoi.models.schemas.models.FractionBoundingSchema attribute) (anemoi.models.schemas.models.HardtanhBoundingSchema attribute) (anemoi.models.schemas.models.ReluBoundingSchema attribute)

versions (anemoi.models.migrations.migrator.MigrationMetadata attribute)

W

window_size (anemoi.models.schemas.decoder.TransformerDecoderSchema attribute) (anemoi.models.schemas.encoder.TransformerEncoderSchema attribute) (anemoi.models.schemas.processor.TransformerProcessorSchema attribute)