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☆779Jul 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
- Serverless GPU API endpoints on Runpod - Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- 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 5 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 4 months ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,628Mar 25, 2022Updated 4 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆134Jul 2, 2019Updated 6 years ago
- Serverless GPU API endpoints on Runpod - Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Implementation of Adversarial Variational Optimization in PyTorch☆42Aug 7, 2018Updated 7 years ago
- Code for visualizing the loss landscape of neural nets☆3,167Apr 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,380Mar 1, 2024Updated 2 years ago
- Relative gradient optimization of the Jacobian term in unsupervised deep learning, NeurIPS 2020☆21Apr 27, 2021Updated 4 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- The differentiable OSQP solver layer for PyTorch☆65Dec 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
- ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning☆285Feb 27, 2023Updated 3 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- PyTorch implementations of algorithms for density estimation☆586May 13, 2021Updated 4 years ago
- Measurements of Three-Level Hierarchical Structure in the Outliers in the Spectrum of Deepnet Hessians (ICML 2019)☆16Apr 27, 2019Updated 6 years ago
- Code for "Supermasks in Superposition"☆126Oct 3, 2023Updated 2 years ago
- PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks☆469Aug 28, 2024Updated last year
- Code for Neural Spline Flows paper☆289Oct 19, 2025Updated 5 months ago
- Official Code for Invertible Residual Networks☆536Jul 28, 2024Updated last year
- Convolutional Neural Tangent Kernel☆112Oct 22, 2019Updated 6 years ago