BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
☆606Nov 28, 2025Updated 3 months ago
Alternatives and similar repositories for backpack
Users that are interested in backpack are comparing it to the libraries listed below
Sorting:
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Jul 19, 2023Updated 2 years ago
- Limitations of the Empirical Fisher Approximation☆49Mar 3, 2025Updated last year
- This repository is no longer maintained. Check☆81Apr 23, 2020Updated 5 years ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆192Dec 5, 2024Updated last year
- ☆33May 21, 2020Updated 5 years ago
- Laplace approximations for Deep Learning.☆537Apr 22, 2025Updated 11 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 3 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆149Oct 1, 2023Updated 2 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,437Aug 21, 2025Updated 7 months ago
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆153Jun 22, 2023Updated 2 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆133Jul 2, 2019Updated 6 years ago
- Distributed K-FAC preconditioner for PyTorch☆95Mar 11, 2026Updated last week
- Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations☆21Mar 25, 2023Updated 2 years ago
- Probabilistic Numerics in Python.☆459Jul 3, 2025Updated 8 months ago
- Efficient PyTorch Hessian eigendecomposition tools!☆386Feb 29, 2024Updated 2 years ago
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆488Jul 1, 2022Updated 3 years ago
- Fast and Easy Infinite Neural Networks in Python☆2,377Mar 1, 2024Updated 2 years ago
- [ICML 2024] SINGD: KFAC-like Structured Inverse-Free Natural Gradient Descent (http://arxiv.org/abs/2312.05705)☆24Nov 4, 2024Updated last year
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆63Updated this week
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆221Updated this week
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆12May 8, 2023Updated 2 years ago
- Figure sizes, font sizes, fonts, and more configurations at minimal overhead. Fix your journal papers, conference proceedings, and other …☆736Jan 29, 2026Updated last month
- A highly efficient implementation of Gaussian Processes in PyTorch☆3,850Mar 12, 2026Updated last week
- ☆627Jan 23, 2026Updated last month
- ☆30Feb 11, 2021Updated 5 years ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆184Jul 17, 2021Updated 4 years ago
- PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks☆778Jul 10, 2025Updated 8 months ago
- An implementation of KFAC for TensorFlow☆199Feb 11, 2022Updated 4 years ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,162Feb 6, 2026Updated last month
- [NeurIPS'19] Deep Equilibrium Models☆794Jul 4, 2022Updated 3 years ago
- A PyTorch implementation of L-BFGS.☆626Feb 21, 2023Updated 3 years ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆317Updated this week
- torch-optimizer -- collection of optimizers for Pytorch☆3,167Mar 22, 2024Updated 2 years ago
- Bayesian optimization in PyTorch☆3,481Updated this week
- Official Code for Invertible Residual Networks☆536Jul 28, 2024Updated last year
- Convolutions and more as einsum for PyTorch☆17Jun 6, 2024Updated last year
- ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning☆285Feb 27, 2023Updated 3 years ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,030Dec 17, 2025Updated 3 months ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,430Feb 20, 2026Updated last month