acerbilab / amortized-conditioning-engineLinks
Amortized Probabilistic Conditioning for Optimization, Simulation and Inference (Chang et al., AISTATS 2025)
☆21Updated 4 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:
- IVON optimizer for neural networks based on variational learning.☆72Updated 11 months ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆108Updated last year
- ☆192Updated last week
- ☆154Updated 3 years ago
- Agustinus' very opiniated publication-ready plotting library☆69Updated 5 months ago
- Sketched linear operations for PyTorch☆92Updated last week
- 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, ...)☆49Updated last week
- Code for our paper: Online Variational Filtering and Parameter Learning☆18Updated 3 years ago
- Parameter-Free Optimizers for Pytorch☆131Updated last year
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆58Updated last month
- Normalizing flows in PyTorch☆410Updated last week
- Likelihood-free AMortized Posterior Estimation with PyTorch☆131Updated last year
- Bayesian inference with Python and Jax.☆34Updated 2 years ago
- Code for efficiently sampling functions from GP(flow) posteriors☆73Updated 4 years ago
- Add a tqdm progress bar to your JAX scans and loops.☆117Updated 5 months ago
- Official Implementation of "Transformers Can Do Bayesian Inference", the PFN paper☆237Updated last year
- Bayesian active learning with EPIG data acquisition☆34Updated last month
- Deep Adaptive Design: Amortizing Sequential Bayesian Experimental Design☆37Updated 4 years ago
- Normalizing Flows with a resampled base distribution☆47Updated 3 years ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆47Updated 9 months ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Updated 2 years ago
- Implicit Deep Adaptive Design (iDAD): Policy-Based Experimental Design without Likelihoods☆21Updated 3 years ago
- Simulation-based inference benchmark☆103Updated 9 months ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆44Updated 4 months ago
- Uncertainty quantification with PyTorch☆375Updated 2 weeks ago
- Normalizing Flows using JAX☆85Updated last year
- Pytorch-like dataloaders for JAX.☆93Updated 5 months ago
- Materials of the Nordic Probabilistic AI School 2023.☆91Updated 2 years ago
- Stainless neural networks in JAX☆34Updated last month