search

google / fedjax

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

External contributions? #47

Open alshedivat opened 3 years ago

alshedivat commented 3 years ago

Hello. Thanks for open sourcing this library! I'm wondering if this repository is open to external contributions? If yes, I'd be interested to send a PR and contribute an implementation of FedPA. Thanks.

stheertha commented 3 years ago

Thanks for being interested in contributing to fedjax. We are in the process of setting up the framework to accept external contributions. Will let you know as soon as we have the necessary pieces in place.

jaehunro commented 3 years ago

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.

Add FedPA 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 FedPA on EMNIST that reproduces the paper results.

Please reach out if you have any additional questions.