p2pfl.learning.frameworks.flax package
======================================

.. automodule:: p2pfl.learning.frameworks.flax
   :members:
   :undoc-members:
   :show-inheritance:

Submodules
----------

.. toctree::
   :maxdepth: 4

   p2pfl.learning.frameworks.flax.flax_dataset
   p2pfl.learning.frameworks.flax.flax_learner
   p2pfl.learning.frameworks.flax.flax_model