GitHub repository: nelaturuharsha/TurboPrune
Primary language: Python
Description: Harness for training/finding lottery tickets in PyTorch. With support for multiple pruning techniques and augmented by distributed training, FFCV and AMP.