acerbilab / amortized-conditioning-engineLinks
Amortized Probabilistic Conditioning for Optimization, Simulation and Inference (Chang et al., AISTATS 2025)
☆21Updated 5 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.☆78Updated last year
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆108Updated last year
- ☆205Updated last month
- Agustinus' very opiniated publication-ready plotting library☆70Updated 7 months ago
- Sketched linear operations for PyTorch☆100Updated last month
- Add a tqdm progress bar to your JAX scans and loops.☆121Updated 7 months ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Updated 2 years ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆131Updated last year
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆94Updated 3 years ago
- Deep Adaptive Design: Amortizing Sequential Bayesian Experimental Design☆39Updated 4 years ago
- ☆155Updated 3 years ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆39Updated 2 months ago
- Generative models with JAX.☆20Updated 11 months ago
- Simulation-based inference benchmark☆107Updated 11 months ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆48Updated 11 months ago
- Pytorch-like dataloaders for JAX.☆97Updated last week
- Gaussian processes in JAX and Flax.☆563Updated last week
- Normalizing flows in PyTorch☆435Updated 3 weeks ago
- Community-sourced list of papers and resources on neural simulation-based inference.☆138Updated last month
- Normalizing Flows using JAX☆85Updated 2 years ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆33Updated 2 years ago
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆61Updated last month
- Laplace Redux -- Effortless Bayesian Deep Learning☆43Updated 6 months ago
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆40Updated 3 years ago
- Simple (and cheap!) neural network uncertainty estimation☆78Updated 2 months ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆298Updated last week
- ☆52Updated 2 years ago
- Official PyTorch implementation of NeuralSVD (ICML 2024)☆20Updated last year
- Code for efficiently sampling functions from GP(flow) posteriors☆74Updated 5 years ago