google/fedjax

Simulation framework for accelerating research in Private Federated Learning

NVIDIA Federated Learning Application Runtime Environment

Plato: A Research Framework for Federated Learning

An open-source framework for machine learning and other computations on decentralized data.

Flower: A Friendly Federated AI Framework

An Open Framework for Federated Learning.

Comprehensive and timely academic information on federated learning (papers, frameworks, datasets, tutorials, workshops)

Minimal and scalable research codebase in JAX, designed for rapid iteration on frontier research in LLM and other autoregressive models.

Flax is a neural network library for JAX that is designed for flexibility.