flax\_extra.model package
=========================

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

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

.. toctree::
   :maxdepth: 1

   flax_extra.model.perceiver
   flax_extra.model.rnn