neural_mass 
-----------

.. automodule:: vbjax.neural_mass
   :members:
   :undoc-members:
   :show-inheritance:

monitor
--------

.. automodule:: vbjax.monitor
   :members:
   :undoc-members:
   :show-inheritance:

loops 
-----

.. automodule:: vbjax.loops
   :members:
   :undoc-members:
   :show-inheritance:

connectome
----------

.. automodule:: vbjax.connectome
   :members:
   :undoc-members:
   :show-inheritance:

layers
------

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




sparse
------

.. automodule:: vbjax.sparse
   :members:
   :undoc-members:
   :show-inheritance:

util
----

.. automodule:: vbjax.util
   :members:
   :undoc-members:
   :show-inheritance:

shtlc
-----

.. automodule:: vbjax.shtlc
   :members:
   :undoc-members:
   :show-inheritance: