acerbilab / amortized-conditioning-engineLinks
Amortized Probabilistic Conditioning for Optimization, Simulation and Inference (Chang et al., AISTATS 2025)
☆21Updated 2 weeks 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.☆81Updated last year
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆109Updated last year
- ☆216Updated last week
- Pytorch-like dataloaders for JAX.☆99Updated last month
- Likelihood-free AMortized Posterior Estimation with PyTorch☆132Updated last year
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆62Updated this week
- Sketched linear operations for PyTorch☆100Updated 3 months ago
- Agustinus' very opiniated publication-ready plotting library☆71Updated 9 months ago
- Simulation-based inference benchmark☆108Updated last year
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆61Updated 5 months ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Updated 2 years ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆50Updated last year
- Normalizing flows in PyTorch☆442Updated 2 months ago
- Uncertainty quantification with PyTorch☆379Updated 2 weeks ago
- Add a tqdm progress bar to your JAX scans and loops.☆124Updated 9 months ago
- ☆156Updated 3 years ago
- Parallel hyperparameter tuning with JAX☆39Updated 6 months ago
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆32Updated 9 months ago
- Gaussian processes in JAX and Flax.☆579Updated last week
- Materials of the Nordic Probabilistic AI School 2022.☆181Updated 3 years ago
- Community-sourced list of papers and resources on neural simulation-based inference.☆146Updated last week
- Simple (and cheap!) neural network uncertainty estimation☆79Updated 4 months ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆311Updated this week
- Code for Gaussian Score Matching Variational Inference☆35Updated 11 months ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆39Updated 3 weeks ago
- Normalizing Flows using JAX☆86Updated 2 years ago
- Generative models with JAX.☆20Updated last year
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆152Updated 2 months ago
- Stainless neural networks in JAX☆34Updated last week