f-dangel / curvlinopsView external linksLinks
PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)
☆62Updated this week
Alternatives and similar repositories for curvlinops
Users that are interested in curvlinops 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
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆191Dec 5, 2024Updated last year
- Bayesian inference on spatial and spatiotemporal data, faster than you can say "Cholesky!"☆12Dec 29, 2025Updated last month
- ☆16Apr 26, 2023Updated 2 years ago
- Laplace approximations for Deep Learning.☆534Apr 22, 2025Updated 9 months ago
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆33Apr 26, 2024Updated last year
- Code for the paper "FedFisher: Leveraging Fisher Information for One-Shot Federated Learning" by Divyansh Jhunjhunwala, Shiqiang Wang, an…☆14Feb 10, 2024Updated 2 years ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆604Nov 28, 2025Updated 2 months ago
- IVON optimizer for neural networks based on variational learning.☆82Nov 7, 2024Updated last year
- Laplace Redux -- Effortless Bayesian Deep Learning☆44Jun 6, 2025Updated 8 months ago
- Source code for my PhD thesis: Backpropagation Beyond the Gradient☆20Feb 25, 2023Updated 2 years ago
- Hessian trace estimation using PyTorch and Hutch++☆20Oct 29, 2020Updated 5 years ago
- Limitations of the Empirical Fisher Approximation☆49Mar 3, 2025Updated 11 months 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☆221Updated this week
- Code for "Functional variational Bayesian neural networks" (https://arxiv.org/abs/1903.05779)☆86Jul 19, 2020Updated 5 years ago
- Efficient PyTorch Hessian eigendecomposition tools!☆384Feb 29, 2024Updated last year
- Multiplicative Normalizing Flows in PyTorch.☆25Nov 3, 2025Updated 3 months ago
- Official Code: Estimating Model Uncertainty of Neural Networks in Sparse Information Form, ICML2020.☆31Mar 3, 2021Updated 4 years ago
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆21Jul 20, 2024Updated last year
- Coverage tests to check the quality of your posterior estimators.☆33Aug 12, 2024Updated last year
- ☆11Nov 13, 2023Updated 2 years ago
- This is unofficial repository for Towards Efficient and Scalable Sharpness-Aware Minimization.☆37Apr 15, 2024Updated last year
- Matrix-free linear algebra in JAX.☆159Dec 2, 2025Updated 2 months ago
- Neural Diffusion Processes☆82Aug 4, 2024Updated last year
- Tutorial materials of the Probabilistic Numerics Spring School.☆35May 10, 2023Updated 2 years ago
- ☆10Apr 5, 2024Updated last year
- Final repository for the project "Unsupervised Machine Learning for the Classification of Astrophysical X-ray Sources". V. S. Pérez-Díaz,…☆10Jan 23, 2024Updated 2 years ago
- Weak lensing measurement pipeline for wide area surveys☆14Apr 14, 2022Updated 3 years ago
- A metrics library for the JAX ecosystem☆40Mar 16, 2023Updated 2 years ago
- Code accompanying paper: Meta-Learning to Improve Pre-Training☆37Nov 2, 2021Updated 4 years ago
- Kernel Localized Linear Regression (KLLR)☆11Jan 31, 2024Updated 2 years ago
- Neural Operators with Applications to the Helmholtz Equation☆11Sep 19, 2024Updated last year
- Implementation of "Federated Full-Parameter Tuning of Billion-Sized Language Models with Communication Cost under 18 Kilobytes" (https://…☆13May 6, 2024Updated last year
- Embedding language models in probability space via log-likelihood vectors☆16Oct 25, 2025Updated 3 months ago
- ☆44Dec 7, 2022Updated 3 years ago
- Integrative Analysis of Patient Health Records and Neuroimages via Memory-based Graph Convolutional Network (ICDM 2018 full paper)☆44May 8, 2019Updated 6 years ago
- Last-layer Laplace approximation code examples☆83Oct 18, 2021Updated 4 years ago
- Contains code for the NeurIPS 2020 paper by Pan et al., "Continual Deep Learning by FunctionalRegularisation of Memorable Past"☆44Nov 10, 2020Updated 5 years ago