Neural Network
==============

* :ref:`learn-nn-modules`

.. _learn-nn-modules:

Modules
-------

.. autoclass:: metatensor.learn.nn.ModuleMap
   :members:

.. autoclass:: metatensor.learn.nn.Sequential
   :members:

.. autoclass:: metatensor.learn.nn.Linear
   :members:

.. autoclass:: metatensor.learn.nn.EquivariantLinear
   :members:

.. autoclass:: metatensor.learn.nn.Tanh
   :members:

.. autoclass:: metatensor.learn.nn.InvariantTanh
   :members:

.. autoclass:: metatensor.learn.nn.ReLU
   :members:

.. autoclass:: metatensor.learn.nn.InvariantReLU
   :members:

.. autoclass:: metatensor.learn.nn.SiLU
   :members:

.. autoclass:: metatensor.learn.nn.InvariantSiLU
   :members:

.. autoclass:: metatensor.learn.nn.LayerNorm
   :members:

.. autoclass:: metatensor.learn.nn.InvariantLayerNorm
   :members: