FedJAX: Federated learning simulation with JAX

Jae Hun Ro, Ananda Theertha Suresh, Ke Wu

Federated learning is a machine learning technique that enables training across decentralized data. Recently, federated learning has become an active area of research due to an inc…