Pruning Neural Networks with Taylor criterion in Pytorch - View it on GitHub