moskomule / hypergrad
Simple and extensible hypergradient for PyTorch
☆16Updated last year
Related projects ⓘ
Alternatives and complementary repositories for hypergrad
- Stochastic Gradient Langevin Dynamics for Bayesian learning☆30Updated 2 years ago
- simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…☆14Updated 4 years ago
- Limitations of the Empirical Fisher Approximation☆45Updated 4 years ago
- Implementation of the PAC Bayesian GP learning method.☆10Updated 5 years ago
- Neural Fixed-Point Acceleration for Convex Optimization☆29Updated 2 years ago
- ☆52Updated 3 months ago
- Adaptive gradient descent without descent☆44Updated 3 years ago
- Implementation of the paper "Meta-Learning by Adjusting Priors Based on Extended PAC-Bayes Theory", Ron Amit and Ron Meir, ICML 2018☆17Updated 3 years ago
- Monotone operator equilibrium networks☆51Updated 4 years ago
- Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561☆25Updated 3 years ago
- Euclidean Wasserstein-2 optimal transportation☆44Updated last year
- Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations☆20Updated last year
- Approximate Inference Turns Deep Networks into Gaussian Processes (dnn2gp)☆48Updated 4 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- Code for Knowledge-Adaptation Priors based on the NeurIPS 2021 paper by Khan and Swaroop.☆16Updated 2 years ago
- Pytorch version of "Deep Convolutional Networks as shallow Gaussian Processes" by Adrià Garriga-Alonso, Carl Rasmussen and Laurence Aitch…☆32Updated 4 years ago
- Implementation of the models and datasets used in "An Information-theoretic Approach to Distribution Shifts"☆25Updated 3 years ago
- Code for Unbiased Implicit Variational Inference (UIVI)☆13Updated 5 years ago
- Code to accompany paper 'Bayesian Deep Ensembles via the Neural Tangent Kernel'☆27Updated 3 years ago
- Supporting code for the paper "Dangers of Bayesian Model Averaging under Covariate Shift"☆33Updated 2 years ago
- Code to accompany the paper Radial Bayesian Neural Networks: Beyond Discrete Support In Large-Scale Bayesian Deep Learning☆33Updated 4 years ago
- An implementation of a differentiable point process and a differentiable spiking neural network.☆20Updated 3 years ago
- ☆29Updated 3 years ago
- ☆15Updated 2 years ago
- Code for "Exponential Family Estimation via Adversarial Dynamics Embedding" (NeurIPS 2019)☆13Updated 4 years ago
- Code repo for "Function-Space Distributions over Kernels"