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.
☆41Jun 12, 2026Updated 2 weeks ago
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 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
- [ACL 2025] OZSpeech: One-step Zero-shot Speech Synthesis with Learned-Prior-Conditioned Flow Matching☆45Feb 9, 2025Updated last year
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- ☆13Mar 21, 2024Updated 2 years ago
- ☆15May 11, 2025Updated last year
- Benchmark evaluating ocean forecasting systems against reference datasets and observations.☆42Updated this week
- Generative models with JAX.☆22Feb 21, 2026Updated 4 months ago
- Clean single-file implementation of offline RL algorithms in JAX☆182Jun 5, 2026Updated 3 weeks ago
- Solving Inverse Problems with Diffusion Optimal Control [NeurIPS 2024]☆19Dec 21, 2024Updated last year
- Implementation of numerous Vision Transformers in Google's JAX and Flax.☆22Aug 30, 2022Updated 3 years ago
- Gradient-informed particle MCMC methods☆12Jan 29, 2024Updated 2 years ago
- Artificial Neural-Net compression and fitting of synthetic spectral grids.☆21Jun 19, 2026Updated last week
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- JAX Implementations of Descript Audio Codec and EnCodec☆37Mar 30, 2025Updated last year
- Regional Ocean Forecasting with Hierarchical Graph Neural Networks☆22Aug 7, 2025Updated 10 months ago
- Average-Reward Reinforcement Learning with Trust Region Methods☆11Oct 17, 2022Updated 3 years ago
- "DeepResearch-Eval: An End-to-End Evaluation Framework for DeepResearch Systems"☆48Oct 16, 2025Updated 8 months ago
- Taylor moment expansion in Python (JaX and SymPy) and Matlab☆11Nov 26, 2024Updated last year
- run tinygrad kernels on esp32☆14Nov 28, 2023Updated 2 years ago
- Stochastic trace estimation using JAX☆19Aug 20, 2025Updated 10 months ago
- ☆28Dec 19, 2025Updated 6 months ago
- Incremental Disentanglement for Environment-Aware Zero-Shot Text-to-Speech Synthesis☆27Mar 21, 2025Updated last year
- 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.
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆161Nov 1, 2022Updated 3 years ago
- [NeurIPS 2025] LabUtopia: High-Fidelity Simulation and Hierarchical Benchmark for Scientific Embodied Agents☆37Jun 16, 2026Updated 2 weeks ago
- Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist☆67Dec 16, 2023Updated 2 years ago
- Implementation of Computer Vision Models in JAX (equinox)☆23Updated this week
- G. Peyré, L. Chizat, F-X. Vialard, J. Solomon, Quantum Optimal Transport for Tensor Field Processing, Arxiv, 2016☆10Apr 13, 2017Updated 9 years ago
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Dec 27, 2023Updated 2 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 9 months ago
- MuChin: A Chinese Colloquial Description Benchmark for Evaluating Language Models in the Field of Music☆26Jan 7, 2026Updated 5 months ago
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆34May 12, 2023Updated 3 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- User-interactive anime colorization☆10Oct 8, 2020Updated 5 years ago
- Official implementation of Categorical Flow Maps on text.☆60Feb 16, 2026Updated 4 months ago
- Speech-To-Text forced-alignment Speech processing Universal PERformance Benchmark☆38May 7, 2025Updated last year
- A language server for bibfile citations☆18Jan 8, 2026Updated 5 months ago
- Neuroevolution is a Competitive Alternative to Reinforcement Learning for Skill Discovery☆17Apr 2, 2026Updated 2 months ago
- （WIP）long form speech generatoins☆31Apr 2, 2025Updated last year
- The companion code for the paper "Variational inference via Wasserstein gradient flows (W-VI) M. Lambert, S. Chewi, F. Bach, S. Bonnabel…☆17Feb 1, 2023Updated 3 years ago