flwrlabs/flower

Flower: A Friendly Federated AI Framework

What it solves

Flower 简化了构建联邦 AI 系统的过程,允许开发者在多个去中心化的设备或服务器上训练机器学习模型,而无需将原始数据移动到中心位置。

How it works

Flower 提供一个作为联邦学习编排层的框架。它被设计为框架无关(framework‑agnostic),意味着可以与任何机器学习库集成——例如 PyTorch、TensorFlow、JAX、Hugging Face 或 scikit‑learn,并且可以扩展以支持自定义策略和通信模式来分布模型训练。

Who it’s for

它面向需要构建可扩展、可定制联邦学习系统的 AI 研究人员和工程师,也适用于希望在边缘设备(如 Android、iOS 或 Raspberry Pi)上实现隐私保护 AI 的开发者。

Highlights

  • Framework Agnostic: Works with almost any ML library, including PyTorch, TensorFlow, Hugging Face, and even NumPy.
  • Highly Customizable: Allows users to override components to create new state-of-the-art federated systems.
  • Broad Device Support: Includes quickstarts for mobile platforms (Android/TFLite, iOS/CoreML) and embedded devices (Raspberry Pi, Nvidia Jetson).
  • Research-Ready: Includes "Flower Baselines," a collection of community-contributed reproductions of popular federated learning publications.