acerbilab / amortized-conditioning-engineLinks
Amortized Probabilistic Conditioning for Optimization, Simulation and Inference (Chang et al., AISTATS 2025)
☆16Updated 2 months ago
Alternatives and similar repositories for amortized-conditioning-engine
Users that are interested in amortized-conditioning-engine are comparing it to the libraries listed below
Sorting:
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆104Updated last year
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆31Updated 2 months ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆103Updated last year
- Normalizing Flows using JAX☆83Updated last year
- Implicit Deep Adaptive Design (iDAD): Policy-Based Experimental Design without Likelihoods☆19Updated 3 years ago
- Neural likelihood-free methods in PyTorch.☆39Updated 5 years ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆35Updated 6 months ago
- Simulation-based inference benchmark☆97Updated 5 months ago
- ☆167Updated last month
- IVON optimizer for neural networks based on variational learning.☆68Updated 7 months ago
- Robust initialisation of inducing points in sparse variational GP regression models.☆33Updated 2 years ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆42Updated 2 weeks ago
- Collecting research materials on neural samplers with diffusion/flow models☆52Updated 3 weeks ago
- All You Need is a Good Functional Prior for Bayesian Deep Learning (JMLR 2022)☆20Updated 3 years ago
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆40Updated 2 months ago
- ☆52Updated 2 years ago
- Generative models with JAX.☆17Updated 5 months ago
- Code for efficiently sampling functions from GP(flow) posteriors☆72Updated 4 years ago
- Bayesian active learning with EPIG data acquisition☆32Updated 2 months ago
- Stainless neural networks in JAX☆34Updated last week
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆100Updated last year
- Deep Adaptive Design: Amortizing Sequential Bayesian Experimental Design☆34Updated 4 years ago
- Minimal Gaussian process library in JAX with a simple (custom) approach to state management.☆12Updated last year
- ☆15Updated 2 years ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆45Updated 5 months ago
- Repository for the work Transforming Gaussian Processes with Normalizing Flows published at AISTATS 2021☆24Updated 2 years ago
- A dataloader, but for JAX☆21Updated last year
- Demos for the paper Generalized Variational Inference (Knoblauch, Jewson & Damoulas, 2019)☆20Updated 6 years ago
- ☆34Updated 2 months ago
- Light-weighted code for Orthogonal Additive Gaussian Processes☆42Updated 10 months ago