Efficient PyTorch Hessian eigendecomposition tools!
☆386Feb 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
Sorting:
- hessian in pytorch☆187Oct 27, 2020Updated 5 years ago
- PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks☆778Jul 10, 2025Updated 8 months ago
- ☆83Jan 15, 2020Updated 6 years ago
- Hessian spectral density estimation in TF and Jax☆125Sep 6, 2020Updated 5 years ago
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆113Mar 3, 2020Updated 6 years ago
- Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations☆21Mar 25, 2023Updated 2 years 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 3 years ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆606Nov 28, 2025Updated 3 months ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆133Jul 2, 2019Updated 6 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆148Oct 1, 2023Updated 2 years ago
- Implementation of Adversarial Variational Optimization in PyTorch☆43Aug 7, 2018Updated 7 years ago
- TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".☆60Jan 12, 2019Updated 7 years ago
- Code for visualizing the loss landscape of neural nets☆3,154Apr 5, 2022Updated 3 years ago
- The differentiable OSQP solver layer for PyTorch☆65Dec 26, 2022Updated 3 years ago
- Implementation and tutorials of normalizing flows with the novel distributions module☆169Aug 13, 2020Updated 5 years ago
- Pytorch implementation of LOLA (https://arxiv.org/abs/1709.04326) using DiCE (https://arxiv.org/abs/1802.05098)☆96Aug 21, 2018Updated 7 years ago
- ☆62Jun 18, 2020Updated 5 years ago
- PyTorch implementations of algorithms for density estimation☆586May 13, 2021Updated 4 years ago
- Mode Connectivity and Fast Geometric Ensembles in PyTorch☆284Oct 24, 2022Updated 3 years ago
- ☆153May 25, 2020Updated 5 years ago
- The Full Spectrum of Deepnet Hessians at Scale: Dynamics with SGD Training and Sample Size☆19May 19, 2019Updated 6 years ago
- ☆15Apr 19, 2020Updated 5 years ago
- Fast and Easy Infinite Neural Networks in Python☆2,377Mar 1, 2024Updated 2 years ago
- Full loss Hessian spectrum approximation tool.☆13Oct 9, 2019Updated 6 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
- Normalizing Flows in Jax☆107Aug 19, 2020Updated 5 years ago
- Continuum Learning with GEM: Gradient Episodic Memory☆406Oct 22, 2018Updated 7 years ago
- ☆33May 21, 2020Updated 5 years ago
- An implementation of shampoo☆78Mar 9, 2018Updated 8 years ago
- Code for Neural Spline Flows paper☆284Oct 19, 2025Updated 4 months ago
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆37Dec 27, 2022Updated 3 years ago
- A probabilistic programming system for simulators and high-performance computing (HPC), based on PyTorch☆392May 8, 2024Updated last year
- PyTorch layer-by-layer model profiler☆606May 23, 2021Updated 4 years ago
- A Generic Tensor-Network library that is designed for quantum simulation, base on the pytorch☆59Jun 21, 2019Updated 6 years ago
- Official Code for Invertible Residual Networks☆536Jul 28, 2024Updated last year
- Relative gradient optimization of the Jacobian term in unsupervised deep learning, NeurIPS 2020☆21Apr 27, 2021Updated 4 years ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆191Dec 5, 2024Updated last year
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,123Apr 20, 2022Updated 3 years ago