Projects
spectral-pruner
A lightweight library for pruning neural networks based on spectral analysis of layer weight matrices.
Python · PyTorch · NumPy
torch-lanczos
GPU-accelerated Lanczos algorithm for computing top-k eigenvalues of the Hessian during training.
Python · PyTorch · CUDA