← all repositories

AshwinRJ/Federated-Learning-PyTorch

A PyTorch implementation of federated learning for distributed training of deep neural networks on decentralized data.

1.4k stars Python ML FrameworksDomain Apps
Federated-Learning-PyTorch
Velocity · 7d
+0.5
★ / day
Trend
steady
star history

This repository implements the seminal federated learning paper for training deep networks from decentralized data. It provides both baseline and federated training approaches on standard datasets (MNIST, Fashion MNIST, CIFAR10) using simple MLP and CNN models. The implementation supports both IID and non-IID data distributions, allowing users to simulate realistic federated scenarios where client data may be heterogeneous.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.