FedJAX is a library for developing custom Federated Learning (FL) algorithms in JAX. - View it on GitHub