A simple and intuitive 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.
☆40Sep 1, 2026Updated last month
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☆31Jul 9, 2024Updated 2 years ago
- Diffusion Probabilistic Model in Jax☆14Apr 20, 2024Updated 2 years ago
- Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist☆66Dec 16, 2023Updated 2 years ago
- Code repository for paper: A score-based filter for nonlinear data assimilation☆14Nov 5, 2024Updated last year
- Code repository for Particle Denoising Diffusion Sampler☆15Apr 2, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…☆156Oct 16, 2025Updated last year
- ☆18Aug 17, 2024Updated 2 years ago
- Jax implementation of a flexible audio loudness meter in Python with implementation of ITU-R BS.1770-4 loudness algorithm☆13Jan 29, 2025Updated last year
- Make Preview great again☆29Oct 5, 2021Updated 5 years ago
- ☆11Sep 8, 2025Updated last year
- Traceable TTS: Toward Watermark-Free TTS with Strong Traceability☆13Sep 4, 2025Updated last year
- ☆13Mar 21, 2024Updated 2 years ago
- GPU-accelerated differentiable sparse linear solver library for JAX☆58Updated this week
- Official Pytorch repository for "Diffusion Posterior Sampling for Linear Inverse Problem Solving: A Filtering Perspective", where FPS (Fi…☆51Aug 6, 2024Updated 2 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- A simple one file Python project to run 2D phase field problems with linear quadrilaterals☆23Sep 1, 2026Updated last month
- Orthogonal polynomials with JAX☆32Updated this week
- Collection of tools to run objective evaluation of speech generation☆14Feb 1, 2026Updated 8 months ago
- Minimal JAX implementation unifying Diffusion and Flow Matching algorithms as alternative strategies for transporting data distributions.☆68Dec 19, 2025Updated 10 months ago
- Minimal multi-gpu implementation of EDM2: "Analyzing and Improving the Training Dynamics of Diffusion Models"☆41Mar 5, 2024Updated 2 years ago
- [ACL 2025] OZSpeech: One-step Zero-shot Speech Synthesis with Learned-Prior-Conditioned Flow Matching☆45Feb 9, 2025Updated last year
- Skeleton for scalable and flexible Jax RL implementations☆101Jul 1, 2023Updated 3 years ago
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆267Sep 4, 2026Updated last month
- ☆14May 9, 2025Updated last year
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Implementation of the ByteDance MagicMix paper☆19Nov 4, 2022Updated 3 years ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆142Dec 20, 2023Updated 2 years ago
- Solve sparse linear systems in JAX using the KLU algorithm☆48Sep 18, 2026Updated last month
- A python code for 2d topology optimization using MMA optimizers in NLOPT☆12Feb 9, 2017Updated 9 years ago
- 3D model for HKUST Redbird (Sundial) built with OpenSCAD☆15Oct 13, 2022Updated 4 years ago
- Actually Sparse Variational Gaussian Processes implemented in GPlow☆11Aug 9, 2023Updated 3 years ago
- Clean single-file implementation of offline RL algorithms in JAX☆184Jun 5, 2026Updated 4 months ago
- ☆11Jun 1, 2025Updated last year
- 2020 NIPA 인공지능 문제해결 경진대회 (예선 1등, 본선 1개 종목 1등)☆11Dec 16, 2020Updated 5 years ago
- 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.
- GMRES algorithm implementation for the numerical solution of an indefinite non-symmetric system of linear equations.☆13Jul 22, 2023Updated 3 years ago
- Generative models with JAX.☆22Feb 21, 2026Updated 8 months ago
- Taylor moment expansion in Python (JaX and SymPy) and Matlab☆11Nov 26, 2024Updated last year
- ☆11Aug 27, 2020Updated 6 years ago
- Official repo for Conditional Injective Flows for Bayesian Imaging☆10Sep 25, 2023Updated 3 years ago
- mcp server for robot and automations☆12Mar 20, 2026Updated 7 months ago
- Chirp instantaneous frequency estimation using stochastic differential equation Gaussian processes☆13Oct 30, 2024Updated 2 years ago