Robustness of Federated Averaging for Non-IID Data
Adam: A Method for Stochastic Optimization