A research library for pytorch-based neural network pruning, compression, and more.

We propose a lossless compression algorithm based on the NTK matrix for DNN. The compressed network yields asymptotically the same NTK as the original (dense and unquantized) network, with its weig…

[NeurIPS 2022] A Fast Post-Training Pruning Framework for Transformers