LICENSE
README.md
setup.py
docs/conf.py
examples/emnist_fed_avg.py
examples/fed_avg.py
examples/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/aggregators/__init__.py
fedjax/aggregators/aggregator.py
fedjax/aggregators/aggregator_test.py
fedjax/aggregators/compression.py
fedjax/aggregators/compression_test.py
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_datasets.py
fedjax/core/client_datasets_benchmark.py
fedjax/core/client_datasets_test.py
fedjax/core/client_samplers.py
fedjax/core/client_samplers_test.py
fedjax/core/dataclasses.py
fedjax/core/federated_algorithm.py
fedjax/core/federated_data.py
fedjax/core/federated_data_test.py
fedjax/core/for_each_client.py
fedjax/core/for_each_client_test.py
fedjax/core/in_memory_federated_data.py
fedjax/core/in_memory_federated_data_test.py
fedjax/core/metrics.py
fedjax/core/metrics_test.py
fedjax/core/models.py
fedjax/core/models_test.py
fedjax/core/optimizers.py
fedjax/core/optimizers_test.py
fedjax/core/regularizers.py
fedjax/core/regularizers_test.py
fedjax/core/serialization.py
fedjax/core/serialization_test.py
fedjax/core/sqlite_federated_data.py
fedjax/core/sqlite_federated_data_test.py
fedjax/core/tree_util.py
fedjax/core/tree_util_test.py
fedjax/core/typing.py
fedjax/core/util.py
fedjax/core/util_test.py
fedjax/datasets/__init__.py
fedjax/datasets/downloads.py
fedjax/datasets/downloads_test.py
fedjax/datasets/emnist.py
fedjax/datasets/emnist_test.py
fedjax/datasets/shakespeare.py
fedjax/datasets/shakespeare_test.py
fedjax/datasets/stackoverflow.py
fedjax/datasets/stackoverflow_test.py
fedjax/experimental/__init__.py
fedjax/experiments/fed_avg/run_fed_avg.py
fedjax/legacy/__init__.py
fedjax/legacy/algorithms/__init__.py
fedjax/legacy/algorithms/agnostic_fed_avg.py
fedjax/legacy/algorithms/agnostic_fed_avg_test.py
fedjax/legacy/algorithms/fed_avg.py
fedjax/legacy/algorithms/fed_avg_test.py
fedjax/legacy/algorithms/hyp_cluster.py
fedjax/legacy/algorithms/hyp_cluster_test.py
fedjax/legacy/algorithms/mime.py
fedjax/legacy/algorithms/mime_lite.py
fedjax/legacy/algorithms/mime_lite_test.py
fedjax/legacy/algorithms/mime_test.py
fedjax/legacy/core/__init__.py
fedjax/legacy/core/client_trainer.py
fedjax/legacy/core/client_trainer_test.py
fedjax/legacy/core/dataclasses.py
fedjax/legacy/core/dataset_util.py
fedjax/legacy/core/dataset_util_test.py
fedjax/legacy/core/evaluation_util.py
fedjax/legacy/core/evaluation_util_test.py
fedjax/legacy/core/federated_algorithm.py
fedjax/legacy/core/metrics.py
fedjax/legacy/core/metrics_test.py
fedjax/legacy/core/model.py
fedjax/legacy/core/model_test.py
fedjax/legacy/core/optimizer.py
fedjax/legacy/core/prefetch.py
fedjax/legacy/core/prefetch_test.py
fedjax/legacy/core/regularizers.py
fedjax/legacy/core/regularizers_test.py
fedjax/legacy/core/serialization.py
fedjax/legacy/core/serialization_test.py
fedjax/legacy/core/test_util.py
fedjax/legacy/core/test_util_test.py
fedjax/legacy/core/tree_util.py
fedjax/legacy/core/tree_util_test.py
fedjax/legacy/core/typing.py
fedjax/legacy/datasets/__init__.py
fedjax/legacy/datasets/emnist.py
fedjax/legacy/datasets/emnist_test.py
fedjax/legacy/datasets/toy_regression.py
fedjax/legacy/datasets/toy_regression_test.py
fedjax/legacy/models/__init__.py
fedjax/legacy/models/emnist.py
fedjax/legacy/models/emnist_test.py
fedjax/legacy/models/toy_regression.py
fedjax/legacy/models/toy_regression_test.py
fedjax/legacy/training/__init__.py
fedjax/legacy/training/checkpoint.py
fedjax/legacy/training/checkpoint_test.py
fedjax/legacy/training/federated_experiment.py
fedjax/legacy/training/federated_experiment_test.py
fedjax/legacy/training/logging.py
fedjax/legacy/training/logging_test.py
fedjax/models/__init__.py
fedjax/models/emnist.py
fedjax/models/emnist_test.py
fedjax/models/shakespeare.py
fedjax/models/shakespeare_test.py
fedjax/models/stackoverflow.py
fedjax/models/stackoverflow_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/structured_flags.py
fedjax/training/structured_flags_test.py
fedjax/training/tasks.py