f-dangel / curvlinopsLinks
PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)
☆40Updated 4 months ago
Alternatives and similar repositories for curvlinops
Users that are interested in curvlinops are comparing it to the libraries listed below
Sorting:
- IVON optimizer for neural networks based on variational learning.☆70Updated 9 months ago
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆215Updated 2 months ago
- Agustinus' very opiniated publication-ready plotting library☆69Updated 3 months ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆107Updated last year
- Supporting code for the paper "Dangers of Bayesian Model Averaging under Covariate Shift"☆33Updated 2 years ago
- Algorithms for computations on random manifolds made easier☆92Updated last year
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆189Updated 8 months ago
- Simple (and cheap!) neural network uncertainty estimation☆69Updated 2 months ago
- Mutual information estimators and benchmark☆51Updated 7 months ago
- Hessian spectral density estimation in TF and Jax☆123Updated 4 years ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆42Updated 2 months ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆147Updated last year
- Normalizing Flows with a resampled base distribution☆47Updated 2 years ago
- Sketched matrix decompositions for PyTorch☆70Updated 2 weeks ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆137Updated 2 months ago
- Repository for the work Transforming Gaussian Processes with Normalizing Flows published at AISTATS 2021☆24Updated 2 years ago
- A primer on Bayesian Neural Networks. The aim of this reading list is to facilitate the entry of new researchers into the field of Bayesi…☆55Updated last year
- This repository contains code for applying Riemannian geometry in machine learning.☆77Updated 4 years ago
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆33Updated 2 years ago
- Laplace approximations for Deep Learning.☆515Updated 4 months ago
- Normalizing Flows using JAX☆84Updated last year
- Demos for the paper Generalized Variational Inference (Knoblauch, Jewson & Damoulas, 2019)☆20Updated 6 years ago
- ☆52Updated 2 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆104Updated last year
- Neural likelihood-free methods in PyTorch.☆39Updated 5 years ago
- ☆243Updated 2 years ago
- PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks☆456Updated 11 months ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Implicit Deep Adaptive Design (iDAD): Policy-Based Experimental Design without Likelihoods☆20Updated 3 years ago
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆34Updated 3 years ago