FedJAX
GoogleJAX-based library for federated learning simulations, built for research ease of use, with federated datasets, models and algorithm implementations, and fast simulation on accelerators.
Not an officially supported Google product, per its README. The last release, 0.0.17, dates from 2023-07-12; the repository was pushed 2026-08-06.
Openness
5 high confidence- license
- Apache-2.0(OSI
- source
- public(the whole library)
- core features withheld
- no — a research library with no paid edition
FedJAX is an Apache-2.0 research library, and no paid tier exists for anything to be held back for.
- https://raw.githubusercontent.com/google/fedjax/HEAD/LICENSE recorded 2026-09-27
The LICENSE body: the verbatim Apache License, Version 2.0 text with no added clause.
- https://raw.githubusercontent.com/google/fedjax/HEAD/README.md recorded 2026-09-27
README: "FedJAX is not an officially supported Google product"; "a JAX-based open source library for Federated Learning simulations that emphasizes ease-of-use in research".
- https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/google%2Ffedjax recorded 2026-09-27
Repository record for google/fedjax: license apache-2.0, not archived, not a fork, pushed 2026-08-06, 272 stars.
- https://ungh.cc/repos/google/fedjax/files/main recorded 2026-09-27
The 140-file tree of google/fedjax at `main`: one root LICENSE, and no ee/, enterprise/, commercial/, proprietary/ or premium/ path anywhere in it. Path absence only, paired with the statement cited beside it.
Adoption
1 medium confidencePyPI downloads of `fedjax`, the library itself.
- https://pypistats.org/api/packages/fedjax/recent recorded 2026-09-27
last_month = 95 downloads for the package `fedjax`.
Capability
2 high confidenceFedJAX simulates federations for research and offers no way to deploy one, one step below Plato, which can run over a real network.
- https://raw.githubusercontent.com/google/fedjax/HEAD/README.md recorded 2026-09-27
README: "FedJAX: Federated learning simulation with JAX"; "a JAX-based open source library for Federated Learning simulations that emphasizes ease-of-use in research".
Verified 2026-09-27