Implementing SCAFFOLD

[Saipraneet]

It might be a good idea to have an implementation of SCAFFOLD as well in the algorithms. I think this can be done by modifying the existing Mime implementation.

[stheertha]

This is a good suggestion, will keep in the radar. We are also planning to accept community contributions. Please let us know if you are interested in implementing it.

[jaehunro]

Thanks for your patience.

We are now ready to accept community contributions. We recommend reading through the contributing guidelines as well as the developer workflow.

If you're interested in implementing, add SCAFFOLD to fedjax.algorithms. Please refer to fed_avg.py and fed_avg_test.py for an example internal implementation and unit tests for FedAvg. Additionally, as mentioned in the contributing guidelines, please also add an associated binary that runs SCAFFOLD on EMNIST that reproduces the paper results.

Please reach out if you have any additional questions.