noether.modeling.models.transolver¶

Classes¶

`TransolverConfig`	Configuration for a Transolver model.
`TransolverPlusPlusConfig`	Configuration for a Transolver++ model.
`Transolver`	Implementation of the Transolver model.

Module Contents¶

class noether.modeling.models.transolver.TransolverConfig(/, **data)¶

Bases: noether.modeling.models.transformer.TransformerConfig, noether.core.models.base.ModelBaseConfig

Configuration for a Transolver model.

Create a new model by parsing and validating input data from keyword arguments.

Raises [ValidationError][pydantic_core.ValidationError] if the input data cannot be validated to form a valid model.

self is explicitly positional-only to allow self as a field name.

Parameters:: data (Any)

model_config¶: Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

attention_arguments: dict¶

set_attention_constructor()¶

Set attention_constructor in transformer_block_config based on data_specs.

Return type:: TransolverConfig

class noether.modeling.models.transolver.TransolverPlusPlusConfig(/, **data)¶

Bases: TransolverConfig

Configuration for a Transolver++ model.

Create a new model by parsing and validating input data from keyword arguments.

Raises [ValidationError][pydantic_core.ValidationError] if the input data cannot be validated to form a valid model.

self is explicitly positional-only to allow self as a field name.

Parameters:: data (Any)

set_attention_constructor()¶

Set attention_constructor in transformer_block_config based on data_specs.

Return type:: TransolverPlusPlusConfig

class noether.modeling.models.transolver.Transolver(config)¶

Bases: noether.modeling.models.transformer.Transformer

Implementation of the Transolver model. Reference code: https://github.com/thuml/Transolver/ Paper: https://arxiv.org/abs/2402.02366 Transolver is a Transformer with a special physics attention mechanism. Hence, we extend the Transformer class, and configure it accordingly.

Parameters:: config (TransolverConfig) – Configuration of the Transolver model.