API Reference ============= .. toctree:: :maxdepth: 3 flax.linen flax.serialization flax.core.frozen_dict flax.struct flax.jax_utils flax.traceback_util flax.traverse_util flax.training flax.config flax.errors