p2pfl.learning.frameworks.flax package

Flax integration for P2PFL.

Submodules

• p2pfl.learning.frameworks.flax.flax_dataset module
  • FlaxExportStrategy
    • FlaxExportStrategy.export()
• p2pfl.learning.frameworks.flax.flax_learner module
  • FlaxLearner
    • FlaxLearner.evaluate()
    • FlaxLearner.fit()
    • FlaxLearner.flax_model
    • FlaxLearner.get_framework()
    • FlaxLearner.interrupt_fit()
    • FlaxLearner.train_step()
• p2pfl.learning.frameworks.flax.flax_model module
  • FlaxModel
    • FlaxModel.build_copy()
    • FlaxModel.decode_parameters()
    • FlaxModel.encode_parameters()
    • FlaxModel.get_framework()
    • FlaxModel.get_parameters()
    • FlaxModel.set_parameters()
  • MLP
    • MLP.hidden_sizes
    • MLP.name
    • MLP.out_channels
    • MLP.parent
    • MLP.scope