Federated Optimization in Heterogeneous Networks (MLSys '20) - View it on GitHub