noether.core.schemas.models.transolver

Classes

TransolverConfig	Configuration for a Transolver model.
TransolverPlusPlusConfig	Configuration for a Transolver++ model.

Module Contents

class noether.core.schemas.models.transolver.TransolverConfig(/, **data)

Bases: noether.core.schemas.models.transformer.TransformerConfig, noether.core.schemas.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_constructor: Literal['transolver', 'transolver_plusplus'] = 'transolver'

Constructor of the attention module. Defaults to ‘dot_product’.

attention_arguments: dict

Additional arguments for the attention module that are only needed for a specific attention implementation.

class noether.core.schemas.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)

attention_constructor: Literal['transolver_plusplus'] = 'transolver_plusplus'

Constructor of the attention module. Defaults to ‘dot_product’.