mindscience.models
===================

GraphCast Models
-----------------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindscience.models.GraphCast.GraphCastNet

Attention Modules
-----------------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindscience.models.transformer.Attention
    mindscience.models.transformer.MultiHeadAttention
    mindscience.models.transformer.TransformerBlock

Vision Transformer (ViT)
-------------------------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindscience.models.transformer.VisionTransformer

Activation Functions
---------------------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindscience.models.layers.get_activation
    mindscience.models.layers.activation.SReLU

Basic Blocks
-------------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindscience.models.layers.DropPath
    mindscience.models.layers.FCSequential
    mindscience.models.layers.InputScale
    mindscience.models.layers.LinearBlock
    mindscience.models.layers.MultiScaleFCSequential
    mindscience.models.layers.ResBlock

UNet2D
------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindscience.models.layers.UNet2D

Mask Layers
-----------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindscience.models.layers.MaskedLayerNorm

Fourier Neural Operators (FNO)
------------------------------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindscience.models.neural_operator.FNO1D
    mindscience.models.neural_operator.FNO2D
    mindscience.models.neural_operator.FNO3D
    mindscience.models.neural_operator.FNOBlocks

Fast Fourier Neural Operators (FFNO)
------------------------------------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindscience.models.neural_operator.FFNO
    mindscience.models.neural_operator.FFNO1D
    mindscience.models.neural_operator.FFNO2D
    mindscience.models.neural_operator.FFNO3D
    mindscience.models.neural_operator.FFNOBlocks