A simple, easy-to-understand library for diffusion models using Flax and Jax. Includes detailed notebooks on DDPM, DDIM, and EDM with simplified mathematical explanations. Made as part of my journey for learning and experimenting with generative AI.
☆41Apr 4, 2026Updated this week
Alternatives and similar repositories for FlaxDiff
Users that are interested in FlaxDiff are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Implementation of Diffusion Transformers and Rectified Flow in Jax☆27Jul 9, 2024Updated last year
- Diffusion Probabilistic Model in Jax☆13Apr 20, 2024Updated last year
- Code repository for Particle Denoising Diffusion Sampler☆15Apr 2, 2024Updated 2 years ago
- Official repository for the Monte Carlo guided Diffusion for Bayesian linear inverse problems paper:☆21Dec 16, 2024Updated last year
- Official Pytorch repository for "Diffusion Posterior Sampling for Linear Inverse Problem Solving: A Filtering Perspective", where FPS (Fi…☆50Aug 6, 2024Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting with the flexibility to host WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Cloudways by DigitalOcean.
- This repository contains the Julia code for the paper "Competitive Gradient Descent"☆25Dec 18, 2019Updated 6 years ago
- [ACL 2025] OZSpeech: One-step Zero-shot Speech Synthesis with Learned-Prior-Conditioned Flow Matching☆45Feb 9, 2025Updated last year
- ☆15May 11, 2025Updated 10 months ago
- Clean single-file implementation of offline RL algorithms in JAX☆174Nov 24, 2025Updated 4 months ago
- Generative models with JAX.☆21Feb 21, 2026Updated last month
- Solving Inverse Problems with Diffusion Optimal Control [NeurIPS 2024]☆19Dec 21, 2024Updated last year
- Next generation MCMC samplers with automatic differentiaion and adaptive Poisson thinning☆13Apr 1, 2026Updated last week
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆142Dec 20, 2023Updated 2 years ago
- 🚀 Official code for “XStreamVGGT: Extremely Memory-Efficient Streaming Vision Geometry Grounded Transformer with KV Cache Compression”, …☆41Jan 27, 2026Updated 2 months ago
- Wordpress hosting with auto-scaling on Cloudways • AdFully Managed hosting built for WordPress-powered businesses that need reliable, auto-scalable hosting. Cloudways SafeUpdates now available.
- Gradient-informed particle MCMC methods☆12Jan 29, 2024Updated 2 years ago
- Artificial Neural-Net compression and fitting of synthetic spectral grids.☆21Nov 14, 2025Updated 4 months ago
- JAX Implementations of Descript Audio Codec and EnCodec☆35Mar 30, 2025Updated last year
- Curated list of JAX Resources and Packages☆38Mar 2, 2026Updated last month
- ☆22Dec 19, 2025Updated 3 months ago
- Average-Reward Reinforcement Learning with Trust Region Methods☆11Oct 17, 2022Updated 3 years ago
- Taylor moment expansion in Python (JaX and SymPy) and Matlab☆11Nov 26, 2024Updated last year
- Incremental Disentanglement for Environment-Aware Zero-Shot Text-to-Speech Synthesis☆27Mar 21, 2025Updated last year
- Fast semantic search for biorXiv manuscripts☆12Feb 16, 2025Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting with the flexibility to host WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Cloudways by DigitalOcean.
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆160Nov 1, 2022Updated 3 years ago
- [NeurIPS 2025] LabUtopia: High-Fidelity Simulation and Hierarchical Benchmark for Scientific Embodied Agents☆32Apr 1, 2026Updated last week
- A language server for bibfile citations☆16Jan 8, 2026Updated 3 months ago
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Dec 27, 2023Updated 2 years ago
- GPFlow based implementation of temporal parallellization of state space GPs☆16Sep 8, 2022Updated 3 years ago
- The open source code of ALMTokenizer2: Towards Low bit-rate and Semantic-rich Audio Tokenizer with Flow-based Scalar Diffusion Transforme…☆45Sep 5, 2025Updated 7 months ago
- USD models for NVIDIA Isaac Sim☆16Aug 8, 2025Updated 8 months ago
- parallel differential expression for single-cell perturbation sequencing☆25Apr 2, 2026Updated last week
- MuChin: A Chinese Colloquial Description Benchmark for Evaluating Language Models in the Field of Music☆26Jan 7, 2026Updated 3 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Speech-To-Text forced-alignment Speech processing Universal PERformance Benchmark☆36May 7, 2025Updated 11 months ago
- The companion code for the paper "Variational inference via Wasserstein gradient flows (W-VI) M. Lambert, S. Chewi, F. Bach, S. Bonnabel…☆14Feb 1, 2023Updated 3 years ago
- Official implementation of Categorical Flow Maps on text.☆50Feb 16, 2026Updated last month
- Neuroevolution is a Competitive Alternative to Reinforcement Learning for Skill Discovery☆17Apr 2, 2026Updated last week
- MIONet: Learning multiple-input operators via tensor product☆46Nov 13, 2022Updated 3 years ago
- Unofficial Implementation of Null-text Inversion (https://arxiv.org/abs/2211.09794)☆12Nov 20, 2022Updated 3 years ago
- A Declarative Language for Expressing Partial World Knowledge to Reinforcement Learning Agents☆17Jan 19, 2024Updated 2 years ago