PyTorch Code for the paper "NUQSGD: Provably Communication-efficient Data-parallel SGD via Nonuniform Quantization"
Primary LanguageJupyter Notebook