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:
- ☆198Updated 3 weeks ago
- IVON optimizer for neural networks based on variational learning.☆72Updated last year
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆108Updated last year
- Simulation-based inference benchmark☆106Updated 10 months ago
- ☆155Updated 3 years ago
- Sketched linear operations for PyTorch☆97Updated last month
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Updated 2 years ago
- Agustinus' very opiniated publication-ready plotting library☆69Updated 6 months ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆131Updated last year
- Code for efficiently sampling functions from GP(flow) posteriors☆73Updated 5 years ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆47Updated 10 months ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆146Updated last week
- Deep Adaptive Design: Amortizing Sequential Bayesian Experimental Design☆38Updated 4 years ago
- Normalizing Flows using JAX☆85Updated last year
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆59Updated 3 weeks ago
- Gaussian processes in JAX and Flax.☆554Updated this week
- Materials of the Nordic Probabilistic AI School 2022.☆180Updated 3 years ago
- Bayesian active learning with EPIG data acquisition☆34Updated 2 months ago
- Uncertainty quantification with PyTorch☆376Updated last month
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Simple (and cheap!) neural network uncertainty estimation☆77Updated last month
- Add a tqdm progress bar to your JAX scans and loops.☆118Updated 6 months ago
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆93Updated 3 years ago
- Normalizing Flows with a resampled base distribution☆47Updated 3 years ago
- Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend☆61Updated last year
- Community-sourced list of papers and resources on neural simulation-based inference.☆135Updated 3 weeks ago
- Code for Gaussian Score Matching Variational Inference☆35Updated 9 months ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆44Updated 5 months ago
- Pre-trained Gaussian processes for Bayesian optimization☆97Updated 7 months ago
- Pytorch-like dataloaders for JAX.☆96Updated 5 months ago