Neural Network Modules
======================

The nn module provides neural network layers and utilities.

Base Classes
------------

.. automodule:: fit.nn.modules.base
   :members:
   :undoc-members:
   :show-inheritance:

Linear Layers
-------------

.. automodule:: fit.nn.modules.linear
   :members:
   :undoc-members:
   :show-inheritance:

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

.. automodule:: fit.nn.modules.activation
   :members:
   :undoc-members:
   :show-inheritance:

Normalization Layers
--------------------

.. automodule:: fit.nn.modules.normalization
   :members:
   :undoc-members:
   :show-inheritance:

Attention Mechanisms
--------------------

.. automodule:: fit.nn.modules.attention
   :members:
   :undoc-members:
   :show-inheritance:

Transformer Components
----------------------

.. automodule:: fit.nn.modules.transformer
   :members:
   :undoc-members:
   :show-inheritance:

Container Modules
-----------------

.. automodule:: fit.nn.modules.container
   :members:
   :undoc-members:
   :show-inheritance:

Functional Interface
--------------------

.. automodule:: fit.nn.functional
   :members:
   :undoc-members:
   :show-inheritance:

Utilities
---------

.. automodule:: fit.nn.utils.model_io
   :members:
   :undoc-members:
   :show-inheritance: