Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs, etc
☆142Dec 20, 2023Updated 2 years ago
Alternatives and similar repositories for minified-generative-models
Users that are interested in minified-generative-models are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141☆26Jan 3, 2024Updated 2 years ago
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆44Mar 1, 2025Updated last year
- A metric for posterior distribution evaluation☆14Jul 15, 2025Updated last year
- Flow-matching algorithms in JAX☆119Aug 12, 2024Updated 2 years ago
- Reversible molecular simulation☆14May 28, 2025Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- ☆14Aug 8, 2023Updated 3 years ago
- A Jax wrapper for cudaKDTree☆11Sep 26, 2025Updated last year
- Exploring how ChatGPT can be used to accelerate research in cosmology.☆13Dec 12, 2022Updated 3 years ago
- ☆14Feb 11, 2025Updated last year
- jax/flax training infrastructure☆16Updated this week
- Curated list of astronomy/astrophysics code packages using JAX☆61Mar 25, 2026Updated 7 months ago
- Steerable E(3) GNN in jax☆24Oct 1, 2023Updated 3 years ago
- Stainless neural networks in JAX☆37Jun 13, 2026Updated 4 months ago
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆31Feb 25, 2025Updated last year
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- A TensorFlow reimplementation of GalSim☆10Mar 8, 2022Updated 4 years ago
- Simulation-based inference in JAX☆55Updated this week
- Custom trained GRACE-OFFF models☆12Updated this week
- Zeus21: Lightning-fast simulations of cosmic dawn☆24Oct 29, 2026Updated last week
- Code of the paper "Listening to the Noise: Blind Denoising with Gibbs Diffusion"☆34Jun 24, 2024Updated 2 years ago
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆26Nov 18, 2025Updated 11 months ago
- Neural Simulation-based Inference with GNN for Jeans Modeling☆10Feb 23, 2024Updated 2 years ago
- Code for the paper "Towards Reliable Simulation-Based Inference with Balanced Neural Ratio Estimation".☆15Nov 14, 2022Updated 3 years ago
- FLOPS counter for all your GPU benchmarking needs☆13Aug 8, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆265Oct 28, 2025Updated last year
- Benchmarking field-level cosmological inference from galaxy surveys.☆16Jul 17, 2025Updated last year
- E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.☆128Oct 20, 2026Updated 2 weeks ago
- ☆22May 7, 2025Updated last year
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆21Mar 10, 2024Updated 2 years ago
- ☆16Jan 23, 2025Updated last year
- ☆246Oct 23, 2026Updated 2 weeks ago
- Metrics for assessing 3D molecular structures.☆11May 9, 2026Updated 5 months ago
- Unleash the power of scheduling☆40Oct 31, 2026Updated last week
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆94May 30, 2024Updated 2 years ago
- Stochastic trace estimation using JAX☆22Aug 27, 2026Updated 2 months ago
- Package for using score-based diffusion model trained on a large variety of survey, and mock survey data.☆11May 15, 2026Updated 5 months ago
- simple simulations for testing weak lensing shear measurement☆14Apr 24, 2026Updated 6 months ago
- E(n) Equivariant GNN in jax☆14Aug 31, 2023Updated 3 years ago
- [ICLR'24] Symphony: Symmetry-Equivariant Point-Centered Spherical Harmonics for Molecule Generation☆30Feb 24, 2025Updated last year
- S2FFT: Differentiable and accelerated spherical transforms☆235Updated this week