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.
☆41Jul 27, 2026Updated last 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 numerous Vision Transformers in Google's JAX and Flax.☆22Aug 30, 2022Updated 3 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
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆22Oct 12, 2023Updated 2 years ago
- Code repository for Particle Denoising Diffusion Sampler☆15Apr 2, 2024Updated 2 years ago
- JAX bindings for Flash Attention v2☆108Feb 28, 2026Updated 5 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Official repository for the Monte Carlo guided Diffusion for Bayesian linear inverse problems paper:☆21Dec 16, 2024Updated last year
- A python module that performas a convolution with a kernel with a variable width☆14Mar 4, 2016Updated 10 years ago
- Official Pytorch repository for "Diffusion Posterior Sampling for Linear Inverse Problem Solving: A Filtering Perspective", where FPS (Fi…☆50Aug 6, 2024Updated 2 years ago
- This repository contains the Julia code for the paper "Competitive Gradient Descent"☆25Dec 18, 2019Updated 6 years ago
- A simple one file Python project to run 2D phase field problems with linear quadrilaterals☆21Aug 25, 2025Updated 11 months ago
- ☆13Mar 5, 2025Updated last year
- GPU-accelerated differentiable sparse linear solver library for JAX☆35Jul 28, 2026Updated last week
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆161Nov 1, 2022Updated 3 years ago
- A Differentiable Topology Optimization Framework for Multiphysics Fluidic Problems☆22Mar 6, 2026Updated 5 months 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.
- [ACL 2025] OZSpeech: One-step Zero-shot Speech Synthesis with Learned-Prior-Conditioned Flow Matching☆45Feb 9, 2025Updated last year
- ☆13Mar 21, 2024Updated 2 years ago
- Implementations of three neural operators and application in Bayesian inference problems☆18Jul 5, 2026Updated last month
- Some useful tools for mesh processing☆12Sep 9, 2025Updated 11 months ago
- Solving Inverse Problems with Diffusion Optimal Control [NeurIPS 2024]☆19Dec 21, 2024Updated last year
- ☆15May 11, 2025Updated last year
- Large-Area Fabrication-Aware Computational Diffractive Optics (SIGGRAPH Asia & TOG 2025)☆17Nov 20, 2025Updated 8 months ago
- Minimal multi-gpu implementation of EDM2: "Analyzing and Improving the Training Dynamics of Diffusion Models"☆41Mar 5, 2024Updated 2 years ago
- ☆33Feb 14, 2024Updated 2 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.
- Clean single-file implementation of offline RL algorithms in JAX☆182Jun 5, 2026Updated 2 months ago
- Artificial Neural-Net compression and fitting of synthetic spectral grids.☆21Jun 19, 2026Updated last month
- Implementation of the ByteDance MagicMix paper☆19Nov 4, 2022Updated 3 years ago
- ⛔️ DEPRECATED GPU Ocean Python/CUDA codebase☆11Nov 9, 2023Updated 2 years ago
- PDE Preserved Neural Network☆58May 15, 2025Updated last year
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆266Mar 21, 2025Updated last year
- Minimal (truly) muP implementation, consistent with TP4 and TP5 papers notation☆14Jan 2, 2026Updated 7 months ago
- Optimal Transport for Machine Learners☆84Updated this week
- Official implementation: "AudioSet-R: A Refined AudioSet with Multi-Stage LLM Label Reannotation"☆20Oct 9, 2025Updated 10 months 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.
- PyTorch Implementation of DSB for Score Based Generative Modeling. Experiments managed using Hydra.☆211Nov 23, 2021Updated 4 years ago
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆39Mar 18, 2025Updated last year
- Regional Ocean Forecasting with Hierarchical Graph Neural Networks☆22Aug 7, 2025Updated last year
- ☆22Apr 21, 2023Updated 3 years ago
- Examples on how to use Fortran and C++ mex files with Matlab☆12Jul 7, 2017Updated 9 years ago
- A C++ Finite Element Analysis software based on the Jem-Jive library☆16Dec 10, 2025Updated 7 months ago
- ☆43Jun 29, 2025Updated last year