Diffusion models in PyTorch
☆128Feb 2, 2026Updated 3 weeks ago
Alternatives and similar repositories for azula
Users that are interested in azula are comparing it to the libraries listed below
Sorting:
- Normalizing flows in PyTorch☆444Dec 1, 2025Updated 3 months ago
- Official implementation of Learning Diffusion Priors from Observations by Expectation Maximization☆34Oct 31, 2025Updated 4 months ago
- Stainless neural networks in JAX☆34Feb 3, 2026Updated 3 weeks ago
- PyTorch-based radio-interferometric imaging reconstruction package with scalable Bayesian uncertainty quantification relying on data-driv…☆12Feb 17, 2025Updated last year
- Repository for Angus H Wright's AstroStatistics Lecture Materials☆15Apr 6, 2022Updated 3 years ago
- An experiment: emcee implemented in JAX☆28Jul 1, 2022Updated 3 years ago
- Differentiable cosmological emulators: the JAX version of CosmoPower☆44Nov 27, 2025Updated 3 months ago
- Exploring how ChatGPT can be used to accelerate research in cosmology.☆12Dec 12, 2022Updated 3 years ago
- A TensorFlow reimplementation of GalSim☆10Mar 8, 2022Updated 3 years ago
- ☆31Jan 18, 2025Updated last year
- Repository with the code of "HNPE: Leveraging Global Parameters for Neural Posterior Estimation"☆14Mar 18, 2024Updated last year
- [UNMAINTAINED] Point spread function modeling tool☆14Jan 22, 2020Updated 6 years ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆133Aug 21, 2024Updated last year
- Flow-matching algorithms in JAX☆115Aug 12, 2024Updated last year
- Sliced Iterative Generator (SIG) & Gaussianizing Iterative Slicing (GIS)☆37Mar 20, 2023Updated 2 years ago
- The galaxy survey codex☆17Jul 12, 2024Updated last year
- Lectures on Bayesian statistics and information theory☆31May 26, 2025Updated 9 months ago
- Simulation-based (likelihood-free) inference customized for astronomical applications☆29Nov 4, 2024Updated last year
- 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
- JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.☆33Feb 22, 2026Updated last week
- Official implementation of "Lost in Latent Space: An Empirical Study of Latent Diffusion Models for Physics Emulation"☆56Oct 8, 2025Updated 4 months ago
- poppy derivatives☆22Aug 13, 2021Updated 4 years ago
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆20Mar 10, 2024Updated last year
- A simple implementation of Hamiltonian Monte Carlo in JAX.☆20Feb 8, 2024Updated 2 years ago
- Probabilistic modeling of tabular data with normalizing flows.☆58Feb 11, 2026Updated 2 weeks ago
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆26Apr 18, 2024Updated last year
- ☆28Oct 7, 2025Updated 4 months ago
- Accelerated Bayesian SED modeling using Amortized Neural Posterior Estimation☆25Dec 11, 2025Updated 2 months ago
- Modern normalizing flows in Python. Simple to use and easily extensible.☆12Feb 11, 2026Updated 2 weeks ago
- ☆10Apr 2, 2024Updated last year
- Likelihood-Free Inference for Gravitational Waves☆42Sep 6, 2021Updated 4 years ago
- Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141☆25Jan 3, 2024Updated 2 years ago
- A system for scientific simulation-based inference at scale.☆167Mar 30, 2024Updated last year
- All-in-one tool to generate, and correctly drizzle, HST, JWST, and Roman PSFs.☆13Feb 19, 2026Updated last week
- tutorials on shape measurement☆10Jan 5, 2022Updated 4 years ago
- Implementation of various equivariant models in JAX☆12Apr 12, 2024Updated last year
- Implementation for Proximal Neural Networks.☆11Sep 17, 2021Updated 4 years ago
- A Jax wrapper for cudaKDTree☆11Sep 26, 2025Updated 5 months ago
- Repository of PIXAR, a Pixel-based Auto-Regressive Language Model☆17Sep 15, 2025Updated 5 months ago