README.md
setup.py
examples/emnist_simple_fed_avg.py
examples/simple_fed_avg.py
examples/simple_fed_avg_test.py
fedjax/__init__.py
fedjax/fedjax_test.py
fedjax/version.py
fedjax.egg-info/PKG-INFO
fedjax.egg-info/SOURCES.txt
fedjax.egg-info/dependency_links.txt
fedjax.egg-info/requires.txt
fedjax.egg-info/top_level.txt
fedjax/algorithms/__init__.py
fedjax/algorithms/agnostic_fed_avg.py
fedjax/algorithms/agnostic_fed_avg_test.py
fedjax/algorithms/fed_avg.py
fedjax/algorithms/fed_avg_test.py
fedjax/algorithms/hyp_cluster.py
fedjax/algorithms/hyp_cluster_test.py
fedjax/algorithms/mime.py
fedjax/algorithms/mime_lite.py
fedjax/algorithms/mime_lite_test.py
fedjax/algorithms/mime_test.py
fedjax/core/__init__.py
fedjax/core/client_trainer.py
fedjax/core/client_trainer_test.py
fedjax/core/dataset_util.py
fedjax/core/dataset_util_test.py
fedjax/core/evaluation_util.py
fedjax/core/evaluation_util_test.py
fedjax/core/federated_algorithm.py
fedjax/core/metrics.py
fedjax/core/metrics_test.py
fedjax/core/model.py
fedjax/core/model_test.py
fedjax/core/optimizer.py
fedjax/core/prefetch.py
fedjax/core/prefetch_test.py
fedjax/core/regularizers.py
fedjax/core/regularizers_test.py
fedjax/core/serialization.py
fedjax/core/serialization_test.py
fedjax/core/test_util.py
fedjax/core/test_util_test.py
fedjax/core/tree_util.py
fedjax/core/tree_util_test.py
fedjax/core/typing.py
fedjax/datasets/__init__.py
fedjax/datasets/emnist.py
fedjax/datasets/emnist_test.py
fedjax/datasets/toy_regression.py
fedjax/datasets/toy_regression_test.py
fedjax/models/__init__.py
fedjax/models/emnist.py
fedjax/models/emnist_test.py
fedjax/models/toy_regression.py
fedjax/models/toy_regression_test.py
fedjax/training/__init__.py
fedjax/training/checkpoint.py
fedjax/training/checkpoint_test.py
fedjax/training/federated_experiment.py
fedjax/training/federated_experiment_test.py
fedjax/training/logging.py
fedjax/training/logging_test.py
fedjax/training/sample.py
fedjax/training/sample_test.py