FAX is a JAX-based library designed for large-scale distributed and federated computations, providing efficient and scalable frameworks for data center federated computations.