Efficient PyTorch Hessian eigendecomposition tools!
☆475Oct 23, 2026Updated 2 weeks 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☆796Jul 10, 2025Updated last year
- ☆85Jan 15, 2020Updated 6 years ago
- hessian in pytorch☆187Oct 27, 2020Updated 6 years ago
- Hessian spectral density estimation in TF and Jax☆125Sep 6, 2020Updated 6 years ago
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆113Mar 3, 2020Updated 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.
- Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations☆22Mar 25, 2023Updated 3 years ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆619Jul 17, 2026Updated 3 months ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,624Mar 25, 2022Updated 4 years ago
- Full loss Hessian spectrum approximation tool.☆13Oct 9, 2019Updated 7 years ago
- Code for visualizing the loss landscape of neural nets☆3,208Apr 5, 2022Updated 4 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆150Oct 1, 2023Updated 3 years ago
- This repository is no longer maintained. Check☆81Apr 23, 2020Updated 6 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆135Jul 2, 2019Updated 7 years ago
- Implementation of Adversarial Variational Optimization in PyTorch☆42Aug 7, 2018Updated 8 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- The Full Spectrum of Deepnet Hessians at Scale: Dynamics with SGD Training and Sample Size☆19May 19, 2019Updated 7 years ago
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆37Dec 27, 2022Updated 3 years ago
- TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".☆61Jan 12, 2019Updated 7 years ago
- A PyTorch implementation of L-BFGS.☆632Feb 21, 2023Updated 3 years ago
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆154Jun 22, 2023Updated 3 years ago
- Fast and Easy Infinite Neural Networks in Python☆2,386Mar 1, 2024Updated 2 years ago
- Code for "On the Relation Between the Sharpest Directions of DNN Loss and the SGD Step Length", ICLR 2019☆11Dec 8, 2022Updated 3 years ago
- Mode Connectivity and Fast Geometric Ensembles in PyTorch☆286Oct 24, 2022Updated 4 years ago
- Python package facilitating the use of Bayesian Deep Learning methods with Variational Inference for PyTorch☆362Oct 12, 2019Updated 7 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- ☆15Apr 19, 2020Updated 6 years ago
- PyTorch implementations of algorithms for density estimation☆589May 13, 2021Updated 5 years ago
- ☆33May 21, 2020Updated 6 years ago
- A differentiation API for PyTorch☆30Jun 21, 2020Updated 6 years ago
- Annealed Importance Sampling (AIS) for generative models.☆16Jul 20, 2018Updated 8 years ago
- Pytorch implementation of LOLA (https://arxiv.org/abs/1709.04326) using DiCE (https://arxiv.org/abs/1802.05098)☆98Aug 21, 2018Updated 8 years ago
- Relative gradient optimization of the Jacobian term in unsupervised deep learning, NeurIPS 2020☆21Apr 27, 2021Updated 5 years ago
- An Implementation of "Small steps and giant leaps: Minimal Newton solvers for Deep Learning" In pytorch☆21Jul 16, 2018Updated 8 years ago
- Code for "Supermasks in Superposition"☆126Oct 3, 2023Updated 3 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☆67Dec 26, 2022Updated 3 years ago
- Normalizing Flows in Jax☆109Aug 19, 2020Updated 6 years ago
- Official Code for Invertible Residual Networks☆535Jul 28, 2024Updated 2 years ago
- Large-batch Training, Neural Network Optimization☆11Nov 8, 2019Updated 7 years ago
- PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks☆474Aug 28, 2024Updated 2 years ago
- PyTorch layer-by-layer model profiler☆604May 23, 2021Updated 5 years ago
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆225Updated this week