jaxsnn.base
===========



.. rubric:: Modules

.. autosummary::
   :toctree:
   :template: custom-module-template.rst
   :recursive:

   jaxsnn.base.compose
   jaxsnn.base.dataset
   jaxsnn.base.params
   jaxsnn.base.root_solving
   jaxsnn.base.types



.. automodule:: jaxsnn.base