Aug, 2021

FedJAX: 使用 JAX 进行联邦学习模拟

TL;DRFedJAX是一个基于JAX的开放源代码库,用于联邦学习模拟,旨在使开发和评估联邦算法更快、更容易,并强调隐私和安全。FedJAX可以在几分钟内使用TPU训练EMNIST数据集上的模型,使用标准超参数在约一小时内训练Stack Overflow数据集。