Models package
==============

Submodules
----------

Models.AFNO module
------------------

.. automodule:: Models.AFNO
   :members:
   :undoc-members:
   :show-inheritance:
   :exclude-members: Mlp, AFNO2D, PatchEmbed, PrecipNet, Block

Models.FNN module
-----------------

.. automodule:: Models.FNN
   :members:
   :undoc-members:
   :show-inheritance:
   
Models.GP module
-----------------

.. automodule:: Models.GP
   :members:
   :undoc-members:
   :show-inheritance:
   :exclude-members: CustomMean, CustomMultitaskMean, UnitGaussianNormalizer
   
Models.GST module
-----------------

.. automodule:: Models.GST
   :members:
   :undoc-members:
   :show-inheritance:
   
Models.VGP\_optimizer module
----------------------------

.. automodule:: Models.VGP_optimizer
   :members:
   :undoc-members:
   :show-inheritance:

Models.neural\_models module
----------------------------

.. automodule:: Models.neural_models
   :members:
   :undoc-members:
   :show-inheritance: