pytorch-hessian-eigenthings
Hessian eigendecomposition tool
An efficient library for computing the eigendecomposition of Hessian matrices in PyTorch models.
Efficient PyTorch Hessian eigendecomposition tools!
367 stars
15 watching
43 forks
Language: Python
last commit: 12 months ago Related projects:
| Repository | Description | Stars |
|---|---|---|
 mariogeiger/hessian | Provides tools for efficient computation of Hessian matrices and Jacobian tensors in PyTorch. | 186 |
 wmvanvliet/pytorch_hmax | An implementation of the HMAX model for vision processing in PyTorch | 57 |
 jmhessel/fmpytorch | PyTorch implementation of a Factorization Machine module | 173 |
 huochaitiantang/pytorch-deep-image-matting | Deep learning implementation of image matting, aiming to separate foreground and background from input images. | 293 |
 pyriemann/pyriemann | A Python package for multivariate data analysis using Riemannian geometry of positive definite matrices | 649 |
 jacobgil/pytorch-pruning | This project provides a PyTorch implementation of pruning techniques to reduce the computational resources required for neural network inference. | 877 |
 alecwangcq/eigendamage-pytorch | A deep learning project implementing structured pruning algorithms in PyTorch for efficient neural network training and inference. | 112 |
 kaiyangzhou/dassl.pytorch | A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. | 1,236 |
 sniklaus/pytorch-extension | A PyTorch CUDA extension that uses CuPy to compute the Hadamard product of two tensors. | 117 |
 ctallec/pyvarinf | A Python package facilitating Bayesian Deep Learning methods with Variational Inference for PyTorch | 359 |
 tianhongdai/integrated-gradient-pytorch | A PyTorch implementation of attributing the impact of inputs on deep neural network outputs | 182 |
 zccyman/pytorch-inference | Enables fast and efficient inference of PyTorch models on Windows 10 platforms using C++ | 89 |
 leviswind/pytorch-transformer | Implementation of a transformer-based translation model in PyTorch | 240 |
 felixgwu/img_classification_pk_pytorch | A PyTorch project for comparing image classification models and facilitating quick experiment setup | 366 |
 gunhochoi/kind-pytorch-tutorial | A comprehensive tutorial on building and training PyTorch models using Python | 392 |