GRAWA: Gradient-based Weighted Averaging for Distributed Training of Deep Learning Models Code-base of the paper GRAWA: Gradient-based Weighted Averaging for Distributed Training of Deep Learning Models accepted to AISTATS 2024. Important: This code base is built on top of LSGD implementation by Yunfei Teng More details to be added...