NVIDIA FLARE (NVIDIA Federated Learning Application Runtime Environment) is a domain-agnostic, open-source, extensible SDK that allows researchers and data scientists to adapt existing ML/DL workflows to a federated paradigm. It enables platform developers to build a secure, privacy-preserving offering for a distributed multi-party collaboration.

FLARE is built on a componentized architecture that allows you to take federated learning workloads from research and simulation to real-world production deployment.

Application Features

• Support both deep learning and traditional machine learning algorithms (eg. PyTorch, TensorFlow, Scikit-learn, XGBoost etc.)
• Support horizontal and vertical federated learning
• Built-in Federated Learning algorithms (e.g., FedAvg, FedProx, FedOpt, Scaffold, Ditto, etc.)
• Support multiple server and client-controlled training workflows (e.g., scatter & gather, cyclic) and validation workflows (global model evaluation, cross-site validation)
• Support both data analytics (federated statistics) and machine learning lifecycle management
• Privacy preservation with differential privacy, homomorphic encryption, private set intersection (PSI)

From Simulation to Real-World

• FLARE Client API to transition seamlessly from ML/DL to FL with minimal code changes
• Simulator and POC mode for rapid development and prototyping
• Fully customizable and extensible components with modular design
• Deployment on cloud and on-premise
• Dashboard for project management and deployment
• Security enforcement through federated authorization and privacy policy
• Built-in support for system resiliency and fault tolerance

Take a look at NVIDIA FLARE Overview for a complete overview, and What's New for the lastest changes.

To install the current release:

$ python3 -m pip install nvflare

You can quickly get started using the FL simulator. A detailed getting started guide is available in the documentation.

Examples and notebook tutorials are located at NVFlare/examples.

We welcome community contributions! Please refer to the contributing guidelines for more details.

Ask and answer questions, share ideas, and engage with other community members at NVFlare Discussions.

Take a look at our growing list of talks, blogs, and publications related to NVIDIA FLARE.

NVIDIA FLARE is released under an Apache 2.0 license.