FAX是使用Jax原语实现联合操作的一个实现。
项目描述
FAX
FAX是一个软件库，旨在将联合编程模型嵌入到JAX中。它旨在简化数据中心中的联合计算编写和执行，特别关注支持大规模联合计算，包括涉及更大模型的计算，并确保它们可以高效运行。FAX使用JAX的映射能力和原语扩展嵌入由TensorFlow Federated定义的原语。
有关如何使用FAX的介绍，请查看tutorials子目录中的colab笔记本。
安装
pip install --upgrade google-fax
构建新的wheel文件
运行python -m build来构建新的google-fax wheel文件。
运行测试
执行run_test.sh脚本。
