Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.
Contributions:291 commits, 8 pushes in 1 year 1 month
deep-learningsparse-matrixtensor
Ok-Topk is a scheme for distributed training with sparse gradients. Ok-Topk integrates a novel sparse allreduce algorithm (less than 6k communication volume which is asymptotically optimal) with the decentralized parallel Stochastic Gradient Descent (SGD) optimizer, and its convergence is proved theoretically and empirically.
Contributions:26 commits, 8 pushes, 4 comments in 6 months
distributed-traininggradient-descentdistributed-deep-learningsparse-allreducetopk-sgd