flax\_extra.layer package
=========================

.. automodule:: flax_extra.layer
   :members:
   :undoc-members:
   :show-inheritance:

Submodules
----------

.. toctree::
   :maxdepth: 1

   flax_extra.layer.io