API Reference
=============

.. toctree::
   :maxdepth: 3

   flax.config
   flax.core.frozen_dict
   flax.errors
   flax.jax_utils
   flax.linen
   flax.serialization
   flax.struct
   flax.traceback_util
   flax.training
   flax.traverse_util