fedjax
fedjax copied to clipboard
google
FedJAX is a JAX-based open source library for Federated Learning simulations that emphasizes ease-of-use in research.
Add a standard implementation of the model for the CIFAR-100 task. The dataset can be found in [fedjax.datasets.cifar100](https://github.com/google/fedjax/blob/main/fedjax/datasets/cifar100.py). For the model architecture, we should follow [“Adaptive Federated Optimization”](https://arxiv.org/abs/2003.00295). The model...
Hi, I'm playing around with clients learning rate but I cannot find a clean way of modifying it. Basically, I need to change the LR following a schedule based on...
I think it would be great to port these datasets from tff to fedjax. I would be happy to make the effort and contribute to the library, but I need...
Hi! congrats on this great library! I've started using it a few days ago and I love it! Is there any way to use a haiku model with a non-trainable...
Hello! I may need to split each client's train dataset into train and validation parts for grid search purposes (for example, tuning the stepsizes in a method). How can this...
Synthetic federated datasets can constructed from standard centralized ones by artificially splitting them among clients. This is usually done using a Dirichlet distribution (e.g. [Hsu et al. 2019](https://arxiv.org/pdf/1909.06335.pdf)). Such synthetic...
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.
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...
Add functionality which a single update step on the clients using their entire batch. This replicates running centralized algorithms on federated datasets.
TensorFlow version: 2.5.3 fedjax version: 0.0.16 jax version: 0.4.8 When I follow the docs (https://fedjax.readthedocs.io/en/latest/fedjax.datasets.html#fedjax.datasets.stackoverflow.load_data) to process the Stackoverflow dataset by using ``` from fedjax.datasets import stackoverflow # Load partially...
