acerbilab / amortized-conditioning-engineLinks
Amortized Probabilistic Conditioning for Optimization, Simulation and Inference (Chang et al., AISTATS 2025)
☆21Updated last week
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:
- IVON optimizer for neural networks based on variational learning.☆80Updated last year
- ☆213Updated this week
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆109Updated last year
- Agustinus' very opiniated publication-ready plotting library☆71Updated 9 months ago
- Simple (and cheap!) neural network uncertainty estimation☆80Updated 3 months ago
- All You Need is a Good Functional Prior for Bayesian Deep Learning (JMLR 2022)☆20Updated 3 years ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆132Updated last year
- ☆156Updated 3 years ago
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆62Updated this week
- Sketched linear operations for PyTorch☆100Updated 3 months ago
- Pytorch-like dataloaders for JAX.☆98Updated last month
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Updated 2 years ago
- Simulation-based inference benchmark☆108Updated last year
- Uncertainty quantification with PyTorch☆379Updated last week
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆32Updated 9 months ago
- Generative models with JAX.☆20Updated last year
- diffusionjax is a simple and accessible diffusion models package in JAX☆50Updated last year
- Deep Adaptive Design: Amortizing Sequential Bayesian Experimental Design☆39Updated 4 years ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆46Updated last year
- Add a tqdm progress bar to your JAX scans and loops.☆124Updated 8 months ago
- Gaussian processes in JAX and Flax.☆579Updated this week
- Bayesian active learning with EPIG data acquisition☆36Updated 5 months ago
- Repo for our paper "Repulsive deep ensembles are Bayesian"☆19Updated 4 years ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆39Updated 2 weeks ago
- Code for Gaussian Score Matching Variational Inference☆35Updated 11 months ago
- Community-sourced list of papers and resources on neural simulation-based inference.☆144Updated last week
- Mutual information estimators and benchmark☆56Updated 4 months ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆44Updated 7 months ago
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆172Updated 3 years ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆152Updated 2 months ago