acerbilab / amortized-conditioning-engineLinks
Amortized Probabilistic Conditioning for Optimization, Simulation and Inference (Chang et al., AISTATS 2025)
☆19Updated last month
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.☆70Updated 9 months ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆105Updated last year
- Agustinus' very opiniated publication-ready plotting library☆67Updated 3 months ago
- ☆177Updated last month
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆104Updated last year
- 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☆130Updated 11 months ago
- Code for efficiently sampling functions from GP(flow) posteriors☆72Updated 4 years ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆42Updated 2 months ago
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆87Updated 3 years ago
- Uncertainty quantification with PyTorch☆367Updated 3 months ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆35Updated last week
- Extensible Tensorflow library for differentiable particle filtering. ICML 2021.☆42Updated 2 years ago
- Pytorch-like dataloaders for JAX.☆94Updated 2 months ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆55Updated 3 weeks ago
- ☆151Updated 2 years ago
- Simple (and cheap!) neural network uncertainty estimation☆68Updated 2 months ago
- Deep Adaptive Design: Amortizing Sequential Bayesian Experimental Design☆35Updated 4 years ago
- ☆52Updated 2 years ago
- Materials of the Nordic Probabilistic AI School 2023.☆90Updated last year
- Simulation-based inference benchmark☆99Updated 6 months ago
- Add a tqdm progress bar to your JAX scans and loops.☆113Updated 2 months ago
- A dataloader, but for JAX☆21Updated last year
- Sketched matrix decompositions for PyTorch☆70Updated this week
- Pre-trained Gaussian processes for Bayesian optimization☆94Updated 3 months ago
- Bayesian inference with Python and Jax.☆34Updated 2 years ago
- Differentiable computations for the signature-PDE-kernel on CPU and GPU.☆55Updated last year
- Official Implementation of "Transformers Can Do Bayesian Inference", the PFN paper☆223Updated 9 months ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆47Updated 6 months ago