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
- This repository is no longer maintained. Check☆81Apr 23, 2020Updated 5 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
- Limitations of the Empirical Fisher Approximation☆49Mar 3, 2025Updated 11 months ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆191Dec 5, 2024Updated last year
- Repository containing Pytorch code for EKFAC and K-FAC perconditioners.☆153Jun 22, 2023Updated 2 years ago
- ☆33May 21, 2020Updated 5 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆133Jul 2, 2019Updated 6 years ago
- Laplace approximations for Deep Learning.☆535Apr 22, 2025Updated 10 months ago
- Distributed K-FAC preconditioner for PyTorch☆95Updated this week
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆148Oct 1, 2023Updated 2 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,436Aug 21, 2025Updated 6 months ago
- Efficient PyTorch Hessian eigendecomposition tools!☆386Feb 29, 2024Updated 2 years ago
- Fast and Easy Infinite Neural Networks in Python☆2,375Mar 1, 2024Updated 2 years ago
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆488Jul 1, 2022Updated 3 years ago
- Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations☆21Mar 25, 2023Updated 2 years ago
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆221Feb 9, 2026Updated 3 weeks ago
- Probabilistic Numerics in Python.☆459Jul 3, 2025Updated 7 months ago
- A highly efficient implementation of Gaussian Processes in PyTorch☆3,834Updated this week
- [NeurIPS'19] Deep Equilibrium Models☆793Jul 4, 2022Updated 3 years ago
- ☆30Feb 11, 2021Updated 5 years ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,158Feb 6, 2026Updated 3 weeks ago
- Official Code for Invertible Residual Networks☆536Jul 28, 2024Updated last year
- PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks☆777Jul 10, 2025Updated 7 months ago
- Bayesian optimization in PyTorch☆3,470Updated this week
- ☆620Jan 23, 2026Updated last month
- torch-optimizer -- collection of optimizers for Pytorch☆3,163Mar 22, 2024Updated last year
- Hessian trace estimation using PyTorch and Hutch++☆20Oct 29, 2020Updated 5 years ago
- A PyTorch implementation of L-BFGS.☆625Feb 21, 2023Updated 3 years ago
- code for "FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Models".☆664Sep 22, 2020Updated 5 years ago
- Last-layer Laplace approximation code examples☆83Oct 18, 2021Updated 4 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
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,401Feb 20, 2026Updated last week
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,027Dec 17, 2025Updated 2 months ago
- Differentiable convex optimization layers☆2,060Updated this week
- A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods☆1,551May 2, 2024Updated last year
- An implementation of KFAC for TensorFlow☆199Feb 11, 2022Updated 4 years ago
- Code for Self-Tuning Networks (ICLR 2019) https://arxiv.org/abs/1903.03088☆61Jun 18, 2019Updated 6 years ago
- code for "Residual Flows for Invertible Generative Modeling".☆273Sep 19, 2023Updated 2 years ago