Diffusion Model Frameworks

Frameworks, implementations, and applications of diffusion models for image, video, and audio generation. Does NOT include general generative AI models, VAEs, GANs, or non-diffusion-based generation approaches.

There are 65 diffusion model frameworks tracked. 1 score above 50 (established tier). The highest-rated is sp-nitech/diffsptk at 59/100 with 196 stars.

Get all 65 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=diffusion-model-frameworks&limit=20"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

#	Framework	Score	Tier	Stars	Language
1	sp-nitech/diffsptk A differentiable version of SPTK	59	Established	196	Python
2	trigeorgis/mdm A TensorFlow implementation of the Mnemonic Descent Method.	48	Emerging	123	Jupyter Notebook
3	clovaai/mxfont Official PyTorch implementation of MX-Font (Multiple Heads are Better than...	47	Emerging	176	Python
4	clovaai/fewshot-font-generation The unified repository for few-shot font generation methods. This repository...	47	Emerging	259	Python
5	Michedev/DDPMs-Pytorch Implementation of various DDPM papers to understand how they work	47	Emerging	87	Python
6	openclimatefix/diffusion_weather Testing out Diffusion-based models for weather and PV forecasting	42	Emerging	18	Dockerfile
7	diStyApps/Stable-Diffusion-Pickle-Scanner-GUI Pickle Scanner GUI	42	Emerging	216	Python
8	jaswon/osu-dreamer a diffusion-based ML model for generating osu! maps from raw audio	41	Emerging	142	Python
9	devstermarts/Notebooks Training RAVE, vschaos2, MSPrior, AFTER and RAVE Latent Diffusion models on...	40	Emerging	35	Jupyter Notebook
10	shehzaidi/pre-training-via-denoising Official implementation of pre-training via denoising for TorchMD-NET	40	Emerging	99	Python
11	tlc121/FsFont Official PaddlePaddle Implementation of Few-Shot Font Generation by Learning...	39	Emerging	84	Python
12	Dexterp37/fontina Data generation, model training and inference for Visual Font Recognition...	39	Emerging	19	Python
13	Smith42/astroddpm A denoising diffusion probabilistic model synthesises galaxies that are...	39	Emerging	58	Shell
14	quickgrid/pytorch-diffusion Implementation of diffusion models in pytorch for custom training.	38	Emerging	32	Python
15	AGhaderi/NDDM Neural drift-diffusion model (NDDM) is a repository to integrate...	38	Emerging	18	Jupyter Notebook
16	geometric-intelligence/pirounet Conditional dance generator	38	Emerging	25	Python
17	peterwilli/sd-leap-booster Fast finetuning using a booster model that puts the initial state to a local minimum	38	Emerging	113	Python
18	DeepWave-KAUST/DMFCO2-pub Official repository for the "A Conditional Diffusion Model for CO₂...	35	Emerging	2	Jupyter Notebook
19	yahiaetman/ms-level-gen Start Small: Training Controllable Game Level Generators without Training...	34	Emerging	9	Jupyter Notebook
20	intel/rho-diffusion Parameterizing n-dimensional density fields using denoising diffusion...	33	Emerging	8	Python
21	hyungting/DGMR-pytorch An implementation of Deep Generative Model of Radars from DeepMind in PyTorch	33	Emerging	45	Python
22	MarcoParola/pytorch-sidu SIDU: SImilarity Difference and Uniqueness method for explainable AI	33	Emerging	46	Python
23	magicxor/picpig Telegram bot working as a frontend for stable-diffusion-webui	33	Emerging	7	C#
24	AngelosNal/Vision-DiffMask Official PyTorch implementation of Vision DiffMask, a post-hoc...	32	Emerging	32	Jupyter Notebook
25	Nikolai10/Diffusion-TF Diffusion models in TensorFlow 2	32	Emerging	5	Python
26	HyeonjeongHa/SYNTHIA Official Pytorch implementations of "SYNTHIA: Novel Concept Design with...	31	Emerging	6	Python
27	rupeshs/SDImageGenerator Stable Diffusion desktop UI for Windows	30	Emerging	25	C++
28	svjack/Stable-Diffusion-Chinese-Extend A fine tune version of Stable Diffusion model on self-translate 10k...	28	Experimental	32	Python
29	hearmeneigh/dataset-rising Toolchain for creating custom datasets and training Stable Diffusion (1.x,...	27	Experimental	18	Python
30	inuwamobarak/img-2-img-generation This repository contains a stable diffusion costumed pipeline for generating...	27	Experimental	7	Jupyter Notebook
31	abdur75648/Echo-DND Official implementation of Echo-DND: A Dual Noise Diffusion Model for Robust...	27	Experimental	8	Python
32	congw729/congw729_log 😉 Log files of reading.	26	Experimental	5	—
33	MrForExample/DC_SSDAE Deep Compression Single-Step Diffusion Autoencoder	24	Experimental	3	Python
34	lazycodedoggy/diffusion-tutorials-cn Diffusion Tutorials (Enhanced Chinese Edition) — A Chinese-translated and...	23	Experimental	1	Jupyter Notebook
35	rlaneth/figment-fusion A Jupyter notebook for AI image generation	23	Experimental	8	Jupyter Notebook
36	rr2203/mnist-diffusion Built a diffusion model on simple handwritten images.	23	Experimental	3	Jupyter Notebook
37	AK2k30/Stable_Diffusion_Model_from_Scratch This project implements a Stable Diffusion model from scratch using PyTorch.	22	Experimental	6	Python
38	HKUST-LongGroup/Diff-II [CVPR 2025] PyTorch implementation of Diff-II	22	Experimental	25	Python
39	johnmartinsson/differentiable-mel-spectrogram The official implementation of DMEL the method presented in the paper "DMEL:...	22	Experimental	22	Python
40	aurelio-amerio/GenSBI-examples Examples and benchmarks for the GenSBI library	22	Experimental	—	Python
41	mountain/metafont-glyphs A metafont-glyphs dataset which facilitate people to define CJK-like glyphs...	21	Experimental	11	Python
42	klae01/ddim-audio Denoising Diffusion Implicit Models	21	Experimental	12	Python
43	gasparyanartur/neurad-diffusion Master Thesis Project - Diffusion Models are used to enhance Neurad Radiance...	20	Experimental	4	Python
44	Eddycrack864/Stable-Diffusion-with-Civitai-and-Hugging-Face-for-Colab Stable Diffusion Automatic 1111 with support for Hugging Face and Civitai...	20	Experimental	8	Jupyter Notebook
45	Shaadalam9/gans-traffic This repository contains the code and analysis for the research paper...	20	Experimental	2	HTML
46	dan-crdll/nn_project_dreamdiffusion Re-implementation of the method proposed in ''DreamDiffusion: Generating...	20	Experimental	8	Jupyter Notebook
47	Yuan-ManX/AIGC-Development-Diffusion This is a library of models for artificial intelligence-generated art,...	19	Experimental	3	Python
48	qtrrb/diffusion A fork of stable diffusion that adds an optimizations, features and an API	19	Experimental	3	Python
49	ES7/Stable-Diffusion-from-Scratch In this repo I have implemented Stable Diffusion from Scratch using PyTorch.	19	Experimental	3	Python
50	YashSharma/AudioDiffusionClassification We use audio diffusion models for improving the classification performance.	19	Experimental	4	Jupyter Notebook
51	ArchieMeng/stable-diffusion-notebookui Colab notebook for quick Stable Diffusion model evaluations.	19	Experimental	4	Jupyter Notebook
52	cure-lab/DiffGuard [ICCV 2023] The official implementation of paper "DiffGuard: Semantic...	19	Experimental	18	Python
53	siddthartha/enso-darknet A simple asynchronuous json API for running Stable Diffusion generation...	18	Experimental	2	Rust
54	HliasOuzounis/SPVD-Diffusion A repository of my Master's Thesis: Accelerating diffusion models for...	17	Experimental	3	Jupyter Notebook
55	RickDYang/mnist-diffusion This repositry is for AI beginers to learn Diffusion model from code scratch.	17	Experimental	1	Python
56	CryAndRRich/npmod Building a simple deep learning framework and some machine learning/deep...	17	Experimental	—	Python
57	aryabro/GhibliDream Few-Shot Image Generation via Diffusion Fine-tuning	13	Experimental	2	Jupyter Notebook
58	alexander-koch/tof2cta.c Port of TOF-MRA to CTA image-to-image translation in C	13	Experimental	—	C
59	Utsab-2010/Diffusion-LM Implementing Diffusion on a Small-Language-Model inspired by the Diffusion-LM paper	13	Experimental	—	Jupyter Notebook
60	lingyzhu0101/Diffusion_Image_Compression [Pre-Print'25] Leveraging Diffusion Knowledge for Realistic Image...	12	Experimental	2	—
61	Arthur-Kitsuragi/FluidDiff A Denoising Diffusion Model for Fluid Field Prediction	12	Experimental	4	Jupyter Notebook
62	zacheryvaughn/softfill-pipelines SoftFill is a Diffusers pipeline based on Differential Diffusion,...	11	Experimental	3	Python
63	Laurent2916/Diffusers.jl 🤗 diffusers, but in Julia (and with worse code) !	11	Experimental	—	Julia
64	aayushmahapatra/stable-diffusion Nextjs application that leverages the model trained by Stability AI and...	11	Experimental	4	TypeScript
65	Nuvotion-Visuals/WebDiffusion Generate batches of images in your browser with Stable Diffusion - private,...	11	Experimental	4	HTML