Flax Tutorial
==============================================

.. toctree::
   :maxdepth: 1
   :caption: Tutorial

   Quickstart_1.md
   Flax_Basics.md
   Managing_Parameters_and_State.md
   Dealing_with_Flax_Module_arguments.md
   Processing_the_entire_Dataset.md
   Batch_normalization.md