FAX: Scalable and Differentiable Federated Primitives in JAX Paper • 2403.07128 • Published Mar 11, 2024 • 13
