This repository is meant to provide a generic code base for neural network pruning, especially for pruning at initialization (PaI). (In preparation now, you may check our survey paper and paper collection below.)
Features
Weight Pruning - Recent advances in pruning neural networks at initialization.