acerbilab / amortized-conditioning-engine
Amortized Probabilistic Conditioning for Optimization, Simulation and Inference (Chang et al., AISTATS 2025)
☆16Updated 3 weeks 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:
- Normalizing Flows using JAX☆83Updated last year
- ☆160Updated this week
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆103Updated last year
- Generative models with JAX.☆17Updated 4 months ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆127Updated 8 months ago
- Neural likelihood-free methods in PyTorch.☆39Updated 5 years ago
- Stainless neural networks in JAX☆34Updated last month
- Code for efficiently sampling functions from GP(flow) posteriors☆72Updated 4 years ago
- Code for Gaussian Score Matching Variational Inference☆33Updated 2 months ago
- Bayesian active learning with EPIG data acquisition☆31Updated 3 weeks ago
- Simulation-based inference benchmark☆96Updated 3 months ago
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆38Updated 3 weeks ago
- Collecting research materials on neural samplers with diffusion/flow models☆46Updated 2 months ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆32Updated 5 months ago
- Robust initialisation of inducing points in sparse variational GP regression models.☆33Updated 2 years ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆100Updated last year
- ☆52Updated 2 years ago
- Deep Adaptive Design: Amortizing Sequential Bayesian Experimental Design☆31Updated 3 years ago
- Implicit Deep Adaptive Design (iDAD): Policy-Based Experimental Design without Likelihoods☆19Updated 3 years ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆30Updated 3 weeks ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆42Updated 3 months ago
- ☆25Updated last year
- IVON optimizer for neural networks based on variational learning.☆65Updated 6 months ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆42Updated 2 years ago
- All You Need is a Good Functional Prior for Bayesian Deep Learning (JMLR 2022)☆21Updated 3 years ago
- Code for our paper: Online Variational Filtering and Parameter Learning☆18Updated 3 years ago
- Jaxpr Visualisation Tool☆28Updated 4 months ago
- ☆33Updated last month
- Add a tqdm progress bar to your JAX scans and loops.☆109Updated this week
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆11Updated 11 months ago