Jae Hun Ro

Hi, Thanks for trying out FedJAX and filing this issue! I think this would mainly be an [optax](https://github.com/deepmind/optax) functionality since FedJAX optimizers just [wrap around existing optax optimizers](https://github.com/google/fedjax/blob/main/fedjax/core/optimizers.py#L29-L65). If you...

Thanks for your patience. We are now ready to accept community contributions. We recommend reading through the [contributing guidelines](https://github.com/google/fedjax/blob/main/docs/contributing.md) as well as the [developer workflow](https://github.com/google/fedjax/blob/main/docs/developer.md). If you're interested in implementing,...

Thanks for filing this! I also think that this will be very useful. A couple of clarifying questions: - What exactly constitutes a "standard numpy dataset"? An iterator of numpy...

Hi there. There hasn't been much work done for checking in a general implementation for this but it would be nice to have. We still actively use and maintain this...

Thanks for your patience. We are now ready to accept community contributions. We recommend reading through the [contributing guidelines](https://github.com/google/fedjax/blob/main/docs/contributing.md) as well as the [developer workflow](https://github.com/google/fedjax/blob/main/docs/developer.md). Add FedPA to [fedjax.algorithms](https://github.com/google/fedjax/tree/main/fedjax/algorithms). Please...

We just checked in an [example of federated averaging with stateful clients](https://github.com/google/fedjax/commit/c4f314bd40bb73c919fbb2e2a42b74a38f4020e6). Similar to the example you linked, it keeps track of number of steps per client across rounds. Some...

Rather than modify [fedjax.training.federated_experiment](https://github.com/google/fedjax/blob/main/fedjax/training/federated_experiment.py), I would suggest writing your own experiment loop since your experiment set up is pretty unique and federated_experiment is pretty straightforward. You should be able to...