flexit package
==============

Subpackages
-----------

flexit.attention
~~~~~~~~~~~~~~~~

.. automodule:: flexit.attention
   :members:
   :show-inheritance:
   :undoc-members:

flexit.attention.positional
~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. automodule:: flexit.attention.positional
   :members:
   :show-inheritance:
   :undoc-members:

flexit.blocks
~~~~~~~~~~~~~

.. automodule:: flexit.blocks
   :members:
   :show-inheritance:
   :undoc-members:

flexit.config
~~~~~~~~~~~~~

.. automodule:: flexit.config
   :members:
   :show-inheritance:
   :undoc-members:

flexit.core
~~~~~~~~~~~

.. automodule:: flexit.core
   :members:
   :show-inheritance:
   :undoc-members:

flexit.factory
~~~~~~~~~~~~~~

.. automodule:: flexit.factory
   :members:
   :show-inheritance:
   :undoc-members:

flexit.inference
~~~~~~~~~~~~~~~~

.. automodule:: flexit.inference
   :members:
   :show-inheritance:
   :undoc-members:

flexit.layers
~~~~~~~~~~~~~

.. automodule:: flexit.layers
   :members:
   :show-inheritance:
   :undoc-members:

flexit.models
~~~~~~~~~~~~~

.. automodule:: flexit.models
   :members:
   :show-inheritance:
   :undoc-members:

flexit.models.heads
~~~~~~~~~~~~~~~~~~~

.. automodule:: flexit.models.heads
   :members:
   :show-inheritance:
   :undoc-members:

flexit.training
~~~~~~~~~~~~~~~

.. automodule:: flexit.training
   :members:
   :show-inheritance:
   :undoc-members:

flexit.utils
~~~~~~~~~~~~

.. automodule:: flexit.utils
   :members:
   :show-inheritance:
   :undoc-members:

Module contents
---------------

.. automodule:: flexit
   :members:
   :show-inheritance:
   :undoc-members: