Efficient PyTorch Hessian eigendecomposition tools!
☆385Feb 29, 2024Updated 2 years ago
Alternatives and similar repositories for pytorch-hessian-eigenthings
Users that are interested in pytorch-hessian-eigenthings are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks☆783Jul 10, 2025Updated 9 months ago
- hessian in pytorch☆187Oct 27, 2020Updated 5 years ago
- ☆83Jan 15, 2020Updated 6 years ago
- Hessian spectral density estimation in TF and Jax☆125Sep 6, 2020Updated 5 years ago
- Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations☆22Mar 25, 2023Updated 3 years ago
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆113Mar 3, 2020Updated 6 years ago
- Full loss Hessian spectrum approximation tool.☆13Oct 9, 2019Updated 6 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆149Oct 1, 2023Updated 2 years ago
- The Full Spectrum of Deepnet Hessians at Scale: Dynamics with SGD Training and Sample Size☆19May 19, 2019Updated 6 years ago
- This repository is no longer maintained. Check☆81Apr 23, 2020Updated 6 years ago
- TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".☆60Jan 12, 2019Updated 7 years ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆611Nov 28, 2025Updated 5 months ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,627Mar 25, 2022Updated 4 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆134Jul 2, 2019Updated 6 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Implementation of Adversarial Variational Optimization in PyTorch☆42Aug 7, 2018Updated 7 years ago
- Code for visualizing the loss landscape of neural nets☆3,175Apr 5, 2022Updated 4 years ago
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆37Dec 27, 2022Updated 3 years ago
- ☆15Apr 19, 2020Updated 6 years ago
- Mode Connectivity and Fast Geometric Ensembles in PyTorch☆285Oct 24, 2022Updated 3 years ago
- Python package facilitating the use of Bayesian Deep Learning methods with Variational Inference for PyTorch☆362Oct 12, 2019Updated 6 years ago
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆153Jun 22, 2023Updated 2 years ago
- Fast and Easy Infinite Neural Networks in Python☆2,382Mar 1, 2024Updated 2 years ago
- Relative gradient optimization of the Jacobian term in unsupervised deep learning, NeurIPS 2020☆21Apr 27, 2021Updated 5 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- The differentiable OSQP solver layer for PyTorch☆66Dec 26, 2022Updated 3 years ago
- Annealed Importance Sampling (AIS) for generative models.☆16Jul 20, 2018Updated 7 years ago
- ☆33May 21, 2020Updated 5 years ago
- Normalizing Flows in Jax☆109Aug 19, 2020Updated 5 years ago
- Pytorch implementation of LOLA (https://arxiv.org/abs/1709.04326) using DiCE (https://arxiv.org/abs/1802.05098)☆98Aug 21, 2018Updated 7 years ago
- ☆153May 25, 2020Updated 5 years ago
- Delta Orthogonal Initialization for PyTorch☆18Jun 27, 2018Updated 7 years ago
- Implementation and tutorials of normalizing flows with the novel distributions module☆169Aug 13, 2020Updated 5 years ago
- PyTorch implementations of algorithms for density estimation☆587May 13, 2021Updated 4 years ago
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- Measurements of Three-Level Hierarchical Structure in the Outliers in the Spectrum of Deepnet Hessians (ICML 2019)☆16Apr 27, 2019Updated 7 years ago
- Code for "Supermasks in Superposition"☆126Oct 3, 2023Updated 2 years ago
- ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning☆286Feb 27, 2023Updated 3 years ago
- PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks☆470Aug 28, 2024Updated last year
- Code for Neural Spline Flows paper☆289Oct 19, 2025Updated 6 months ago
- Official Code for Invertible Residual Networks☆536Jul 28, 2024Updated last year
- Convolutional Neural Tangent Kernel☆112Oct 22, 2019Updated 6 years ago