For a framework for federated learning, the first results are federatedai/fate, adap/flower and project-monai/monai. openmined/pysyft and tensorflow/federated round out the shortlist. Compare the match explanations and check the project documentation against your requirements.
These libraries enable collaborative machine learning model training while keeping sensitive raw data on local devices.
FATE is an open-source federated learning platform that enables multiple organizations to collaboratively train machine learning models without exposing raw data to any party. It provides a complete framework for private data collaboration, allowing participants to jointly compute on sensitive information while maintaining data privacy and security guarantees through secure multi-party computation protocols. The platform distinguishes itself through its comprehensive infrastructure management capabilities, supporting automated deployment of multi-party clusters using Ansible-driven provisioni
FATE is a comprehensive, open-source federated learning platform that lets multiple parties collaboratively train models without sharing raw data, and its support for secure aggregation protocols and multi-party deployment aligns well with the requested features like secure aggregation, cross-silo setup, and simulation.
Flower is a federated learning framework and distributed machine learning orchestrator designed to train models across decentralized devices. It functions as a privacy-preserving toolkit that enables model training and data analysis on local hardware, ensuring raw data remains on the device while contributing to a synchronized global model. The system employs an agnostic wrapper and integrator to connect diverse machine learning libraries, allowing different frameworks to operate within the same training loop. It uses a remote procedure call orchestrator to manage the exchange of model weight
Flower is a purpose-built federated learning framework that orchestrates decentralized training across devices and silos, supports TensorFlow and PyTorch, includes privacy-preserving techniques like secure aggregation and differential privacy, and provides simulation and benchmarking tools — exactly meeting this search's requirements.
MONAI is a PyTorch-based deep learning framework and library specifically designed for healthcare imaging. It provides a suite of domain-specific neural network architectures, specialized loss functions, and preprocessing pipelines tailored for analyzing multi-dimensional medical data. The project distinguishes itself through a decentralized federated learning system that allows models to learn from datasets across multiple institutions without exchanging raw patient images. It also features AI-assisted medical image annotation tools and a standardized model bundling system to ensure consiste
MONAI is a PyTorch-based deep learning framework for healthcare imaging that includes a decentralized federated learning system for training across institutions without sharing raw data — it fits the federated learning category but is specialized for medical imaging and does not explicitly cover TensorFlow support or differential privacy.
PySyft is a privacy-preserving machine learning framework and remote computation engine. It functions as a decentralized data analysis orchestrator that allows for the execution of data science workflows on remote servers without requiring the transfer of raw private data from the host device. The platform provides a secure collaboration environment where data owners manage permissions and authorize specific collaborators to run computations. It differentiates its workflow by utilizing mock data for local development and validation before submitting final analysis jobs to private remote serve
PySyft is a privacy-preserving machine learning framework that supports federated learning and remote computation on decentralized data without sharing raw data, fitting the core need, and it integrates with PyTorch for model training, though TensorFlow support and specific features like simulation/benchmarking are not clearly indicated.
An open-source framework for machine learning and other computations on decentralized data.
TensorFlow Federated is a dedicated framework for federated learning that lets you train models on decentralized data without sharing raw data, with strong TensorFlow integration and support for secure aggregation, differential privacy, cross-device/silo, and simulation, though its primary focus is TensorFlow rather than PyTorch.
This project is a research-oriented platform designed for simulating decentralized machine learning environments. It provides a framework for training models across multiple client nodes while keeping raw data localized, enabling the evaluation of model convergence and performance under various distributed network conditions. The system utilizes a parameter-server architecture to coordinate training, where a central coordinator manages the global model state and aggregates weight updates from distributed participants. By decoupling the training orchestration logic from the underlying neural n
This is a PyTorch-specific implementation of federated learning that lets you train models on decentralized data without sharing raw data, so it fits the category directly, though it lacks broader framework support (e.g., TensorFlow) and advanced features like secure aggregation or differential privacy.
SecretFlow is a privacy computing framework and platform designed for secure multi-party computation, federated learning, and privacy-preserving data analysis across independent nodes. It provides a management system to coordinate secure workloads and cryptographic tasks across a distributed cluster. The project enables joint data analysis and machine learning on partitioned datasets using cryptographic protocols. It allows for the training of models and the execution of analytical queries across multiple parties without exposing raw source information to any single participant. The framewor
SecretFlow is a privacy computing framework that includes federated learning, secure multi-party computation, and differential privacy, enabling training on decentralized data without sharing raw data; it does not explicitly mention integration with TensorFlow or PyTorch, but it squarely belongs to the federated learning framework category.
Pyro is a deep probabilistic programming library and differentiable probabilistic modeler designed for Bayesian inference. It functions as a probabilistic programming language that allows for the construction of complex graphical models using PyTorch tensors and automatic differentiation. The framework enables the definition of universal probabilistic models as standard Python functions. It integrates deep learning with probabilistic modeling to compute posterior distributions and estimate latent variables through gradient-based optimization and algorithmic solvers. The system provides a pro
Pyro is a deep probabilistic programming library for Bayesian inference using PyTorch, but it is not a federated learning framework — it lacks support for decentralized training, secure aggregation, or differential privacy, which are the core capabilities of your search.
This project is a Rust interface for the PyTorch C++ library, serving as a deep learning framework and tensor computing library. It functions as a C++ API wrapper that enables the manipulation of multi-dimensional arrays and the execution of neural network architectures across CPU and GPU hardware accelerators. The library provides a TorchScript inference engine to load and execute just-in-time compiled models. It also supports Rust and Python interoperability, allowing for the creation of Python extensions that share tensor data through a common interface. The system covers deep learning mo
tch-rs is a Rust binding for the PyTorch deep learning library, enabling tensor operations and model training on CPU/GPU, but it contains no federated learning features such as decentralized data handling, secure aggregation, or differential privacy, so it does not serve as a federated learning framework or library.
XGBoost is a distributed machine learning library for implementing scalable gradient boosting decision trees used for regression, classification, and ranking. It functions as a predictive model framework and a cross-language toolkit, providing a core implementation with native bindings for Python, R, Java, Scala, and C++. The system is designed as a GPU-accelerated library that utilizes CUDA and NCCL to speed up the training of decision tree ensembles. It operates as a distributed framework capable of scaling training and prediction across multi-node clusters and GPU environments to process m
XGBoost is a distributed gradient boosting library for tree-based models, not a dedicated federated learning framework; despite a "Federated Learnings" tag, it trains on centralized or shuffled data rather than keeping data decentralized, and it does not provide secure aggregation or differential privacy.
| Repository | Stars | Language | License | Last push |
|---|---|---|---|---|
| federatedai/fate | 6K | Python | apache-2.0 | |
| adap/flower | 7K | Python | Apache-2.0 | |
| project-monai/monai |
| 7.9K |
| Python |
| apache-2.0 |
| openmined/pysyft | 9.9K | Python | Apache-2.0 |
| tensorflow/federated | 2.4K | Python | Apache-2.0 |
| shaoxiongji/federated-learning | 1.5K | Python | MIT |
| secretflow/secretflow | 2.6K | Python | apache-2.0 |
| pyro-ppl/pyro | 9K | Python | Apache-2.0 |
| laurentmazare/tch-rs | 5.3K | Rust | apache-2.0 |
| dmlc/xgboost | 28.5K | C++ | Apache-2.0 |