FedJAX is a JAX-based open source library for Federated Learning simulations that emphasizes ease-of-use in research.
Apache License 2.0
252
stars
41
forks
source link
Add example of stateful federated averaging where a counter tracking the total number of training steps taken for each client is maintained and updated across rounds #293
Add example of stateful federated averaging where a counter tracking the total number of training steps taken for each client is maintained and updated across rounds
NOTE: Keeping track of client state is not recommended for large numbers of clients (i.e. in the cross-device setting) or when each client state is very large.
Add example of stateful federated averaging where a counter tracking the total number of training steps taken for each client is maintained and updated across rounds
NOTE: Keeping track of client state is not recommended for large numbers of clients (i.e. in the cross-device setting) or when each client state is very large.