fedjax

FedJAX is a JAX-based open source library for Federated Learning simulations that emphasizes ease-of-use in research.

GitHub

251 stars
11 watching
42 forks
Language: Python
last commit: about 1 month ago
federated-learningjax