Diffusion models in PyTorch
☆128Mar 8, 2026Updated last week
Alternatives and similar repositories for azula
Users that are interested in azula are comparing it to the libraries listed below
Sorting:
- Implementation of papers in 101 lines of code.☆18Nov 12, 2023Updated 2 years ago
- Normalizing flows in PyTorch☆451Mar 10, 2026Updated last week
- Official implementation of Learning Diffusion Priors from Observations by Expectation Maximization☆35Oct 31, 2025Updated 4 months ago
- Stainless neural networks in JAX☆34Feb 3, 2026Updated last month
- A TensorFlow reimplementation of GalSim☆10Mar 8, 2022Updated 4 years ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆133Aug 21, 2024Updated last year
- An experiment: emcee implemented in JAX☆28Jul 1, 2022Updated 3 years ago
- PyTorch-based radio-interferometric imaging reconstruction package with scalable Bayesian uncertainty quantification relying on data-driv…☆12Feb 17, 2025Updated last year
- ☆24Dec 16, 2024Updated last year
- Repository for Angus H Wright's AstroStatistics Lecture Materials☆16Apr 6, 2022Updated 3 years ago
- Official implementation of "Lost in Latent Space: An Empirical Study of Latent Diffusion Models for Physics Emulation"☆56Oct 8, 2025Updated 5 months ago
- Repository with the code of "HNPE: Leveraging Global Parameters for Neural Posterior Estimation"☆14Mar 18, 2024Updated 2 years ago
- Differentiable cosmological emulators: the JAX version of CosmoPower☆45Nov 27, 2025Updated 3 months ago
- Semantic alignment of astronomical data with natural language using multi-modal models. (Jax) Code associated with https://arxiv.org/abs/…☆17Oct 18, 2024Updated last year
- [UNMAINTAINED] Point spread function modeling tool☆14Jan 22, 2020Updated 6 years ago
- Simulation-based (likelihood-free) inference customized for astronomical applications☆29Nov 4, 2024Updated last year
- The galaxy survey codex☆17Jul 12, 2024Updated last year
- poppy derivatives☆22Aug 13, 2021Updated 4 years ago
- Flow-matching algorithms in JAX☆116Aug 12, 2024Updated last year
- Probabilistic modeling of tabular data with normalizing flows.☆58Feb 11, 2026Updated last month
- Richard @rdgao & Michael @michaeldeistler: using neural network-based regression and density estimation for Generalized Bayesian Inferenc…☆11Dec 19, 2025Updated 3 months ago
- Sliced Iterative Generator (SIG) & Gaussianizing Iterative Slicing (GIS)☆37Mar 20, 2023Updated 3 years ago
- Lectures on Bayesian statistics and information theory☆32Updated this week
- Investigating the information content in the cosmic web through dark matter halo graphs. Arxiv paper: https://arxiv.org/abs/2207.05202☆13Jan 18, 2024Updated 2 years ago
- Implementation of various equivariant models in JAX☆12Apr 12, 2024Updated last year
- Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141☆25Jan 3, 2024Updated 2 years ago
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆20Mar 10, 2024Updated 2 years ago
- Notebook to go along with a lecture for the MIT course 8.16: Data Science in Physics on neural simulation-based inference.☆60Oct 31, 2023Updated 2 years ago
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆26Apr 18, 2024Updated last year
- Code for Gaussian Score Matching Variational Inference☆35Feb 23, 2025Updated last year
- Accelerated Bayesian SED modeling using Amortized Neural Posterior Estimation☆25Dec 11, 2025Updated 3 months ago
- A system for scientific simulation-based inference at scale.☆167Mar 30, 2024Updated last year
- Sampling and visualising the von Mises-Fisher distribution. An important hyper-spherical distribution.☆15Jun 5, 2020Updated 5 years ago
- ☆10Apr 2, 2024Updated last year
- "How to Trust Your Diffusion Models: A Convex Optimization Approach to Conformal Risk Control"☆18Jan 6, 2026Updated 2 months ago
- All-in-one tool to generate, and correctly drizzle, HST, JWST, and Roman PSFs.☆13Mar 14, 2026Updated last week
- A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control…☆45Mar 27, 2025Updated 11 months ago
- Implementation for Proximal Neural Networks.☆11Sep 17, 2021Updated 4 years ago
- ☆31Nov 25, 2025Updated 3 months ago