Last released May 2, 2025
DrJAX - Scalable and Differentiable MapReduce Primitives in JAX.
Last released Sep 17, 2024
An open-source framework for machine learning and other computations on decentralized data.
Last released Feb 29, 2024
FAX is an implementation of federated operations using Jax primitives.
Supported by