Handy PyTorch implementation of Federated Learning (for your painless research)

PyTorch implementation of Federated Learning algorithms FedSGD, FedAvg, FedAvgM, FedIR, FedVC, FedProx and standard SGD, applied to visual classification. Client distributions are synthesized with arbitrary non-identicalness and imbalance (Dirichlet priors). Client systems can be arbitrarily heterogeneous. Several mobile-friendly models are prov…

A secure, scalable, and modular implementation of a Federated Learning system designed for distributed environments. Built using Python, PyTorch, gRPC, and Consul, this project enables multiple clients to collaboratively train machine learning models while ensuring data privacy and dynamic service discovery.