AdamCobb / hamiltorch
PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks
☆438Updated 7 months ago
Alternatives and similar repositories for hamiltorch:
Users that are interested in hamiltorch are comparing it to the libraries listed below
- Laplace approximations for Deep Learning.☆498Updated last month
- ☆240Updated 2 years ago
- Normalizing flows in PyTorch☆372Updated last month
- Manifold-learning flows (ℳ-flows)☆229Updated 4 years ago
- Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.☆434Updated last year
- Code for the paper "Learning Differential Equations that are Easy to Solve"☆278Updated 3 years ago
- Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.☆573Updated last week
- Gaussian processes in JAX.☆489Updated this week
- Normalizing flows in PyTorch☆896Updated 3 months ago
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆305Updated 3 weeks ago
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆169Updated 3 years ago
- ☆151Updated 2 years ago
- Deep GPs built on top of TensorFlow/Keras and GPflow☆124Updated 5 months ago
- Constrained optimization toolkit for PyTorch☆673Updated 3 years ago
- PyTorch implementation of bayesian neural network [torchbnn]☆521Updated 8 months ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆574Updated 3 months ago
- Gaussian process modelling in Python☆222Updated 3 months ago
- sbi is a Python package for simulation-based inference, designed to meet the needs of both researchers and practitioners. Whether you nee…☆661Updated this week
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆891Updated last month
- Official Implementation of "Transformers Can Do Bayesian Inference", the PFN paper☆210Updated 5 months ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆956Updated 2 weeks ago
- Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"☆286Updated 4 years ago
- Code for Neural Spline Flows paper☆266Updated 4 years ago
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆211Updated this week
- A community repository for benchmarking Bayesian methods☆110Updated 3 years ago
- Probabilistic Numerics in Python.☆448Updated 11 months ago
- A library for Bayesian neural network layers and uncertainty estimation in Deep Learning extending the core of PyTorch☆573Updated 2 months ago
- ☆559Updated last week
- Large-scale, multi-GPU capable, kernel solver☆186Updated 8 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆107Updated 3 weeks ago