fedjax

Federated learner simulator

A library that provides an easy-to-use framework for simulating federated learning algorithms

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

GitHub

254 stars
11 watching
41 forks
Language: Python
last commit: 13 days ago
federated-learningjax

Related projects:

Repository Description Stars
dawenzi098/sfl-structural-federated-learning A Python implementation of Personalized Federated Learning with Graph using PyTorch. 49
mediabrain-sjtu/feddisco A federated learning framework with discrepancy-aware collaboration for decentralized data training 68
sungwon-han/fedx An unsupervised federated learning algorithm that uses cross knowledge distillation to learn meaningful data representations from local and global levels. 69
ibm/federated-learning-lib A framework for collaborative distributed machine learning in enterprise environments. 500
federatedai/fate-client Provides tools and APIs for designing, scheduling, and running federated machine learning jobs in a secure and efficient manner. 3
alibaba/federatedscope A comprehensive platform for federated learning, providing an event-driven architecture and flexible customization for various tasks in academia and industry. 1,335
mediabrain-sjtu/pfedgraph This project enables personalized federated learning with inferred collaboration graphs to improve the performance of machine learning models on non-IID (non-independent and identically distributed) datasets. 26
mediabrain-sjtu/fedgela This project enables federated learning across partially class-disjoint data with curated bilateral curation. 11
haozzh/fedcr Evaluates various methods for federated learning on different models and tasks. 19
yuetan031/fedproto An implementation of federated learning with prototype-based methods across heterogeneous clients 134
yamingguo98/fediir An implementation of a federated learning algorithm that generalizes to out-of-distribution scenarios using implicit invariant relationships 10
idanachituve/pfedgp An implementation of Personalized Federated Learning with Gaussian Processes using Python. 32
omarfoq/fedem Develops and evaluates federated learning algorithms for personalizing machine learning models across heterogeneous client data distributions. 157
sap-samples/machine-learning-diff-private-federated-learning Simulates a federated learning setting to preserve individual data privacy 365
federatedai/fate-serving A high-performance serving system for federated learning models, providing support for online algorithms, real-time inference, and model management. 138