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 78 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 78 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	43	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	Smith42/astroddpm A denoising diffusion probabilistic model synthesises galaxies that are...	39	Emerging	58	Shell
13	peterwilli/sd-leap-booster Fast finetuning using a booster model that puts the initial state to a local minimum	38	Emerging	113	Python
14	quickgrid/pytorch-diffusion Implementation of diffusion models in pytorch for custom training.	38	Emerging	32	Python
15	MarcoParola/pytorch-sidu SIDU: SImilarity Difference and Uniqueness method for explainable AI	38	Emerging	46	Python
16	Dexterp37/fontina Data generation, model training and inference for Visual Font Recognition...	35	Emerging	19	Python
17	DeepWave-KAUST/DMFCO2-pub Official repository for the "A Conditional Diffusion Model for CO₂...	33	Emerging	2	Jupyter Notebook
18	hyungting/DGMR-pytorch An implementation of Deep Generative Model of Radars from DeepMind in PyTorch	33	Emerging	45	Python
19	AngelosNal/Vision-DiffMask Official PyTorch implementation of Vision DiffMask, a post-hoc...	32	Emerging	32	Jupyter Notebook
20	Cyr-Ch/Diffusion-101 A beginner-friendly guide to building and training diffusion models from...	31	Emerging	5	Jupyter Notebook
21	geometric-intelligence/pirounet Conditional dance generator	31	Emerging	25	Python
22	AGhaderi/NDDM Neural drift-diffusion model (NDDM) is a repository to integrate...	31	Emerging	18	Jupyter Notebook
23	svjack/Stable-Diffusion-Chinese-Extend A fine tune version of Stable Diffusion model on self-translate 10k...	28	Experimental	32	Python
24	sdkyuanpanda/SuperDiff Implementation of Diffusion Models for Conditional Generation of...	28	Experimental	6	Jupyter Notebook
25	yahiaetman/ms-level-gen Start Small: Training Controllable Game Level Generators without Training...	27	Experimental	9	Jupyter Notebook
26	Ipsedo/MusicDiffusion Music synthesis with diffusion model	26	Experimental	5	Jupyter Notebook
27	Anzhc/Timestep-Attention-and-other-shenanigans Novel approach to scheduling timesteps for diffusion models. And other...	26	Experimental	7	Python
28	magicxor/picpig Telegram bot working as a frontend for stable-diffusion-webui	26	Experimental	7	C#
29	abt8601/diffusion-maps A library for diffusion maps (Numerical software development project)	26	Experimental	5	C++
30	intel/rho-diffusion Parameterizing n-dimensional density fields using denoising diffusion...	26	Experimental	8	Python
31	Nikolai10/Diffusion-TF Diffusion models in TensorFlow 2	25	Experimental	5	Python
32	ebiiim/sd-webui-docker Yet another Docker image for stable-diffusion-webui with Kubernetes support.	25	Experimental	5	Dockerfile
33	HyeonjeongHa/SYNTHIA Official Pytorch implementations of "SYNTHIA: Novel Concept Design with...	24	Experimental	6	Python
34	jaabmar/cp_fuse Implementation for the paper "Copyright-Protected Language Generation via...	24	Experimental	7	Python
35	tr-nukala/pytorch_diffusion A minimal PyTorch diffusion model demo that teaches denoising concepts using...	23	Experimental	1	Jupyter Notebook
36	lazycodedoggy/diffusion-tutorials-cn Diffusion Tutorials (Enhanced Chinese Edition) — A Chinese-translated and...	23	Experimental	1	Jupyter Notebook
37	rupeshs/SDImageGenerator Stable Diffusion desktop UI for Windows	23	Experimental	25	C++
38	hemangchawla/DiffusionModelSprites Deeplearning.ai \| How Diffusion Models Work Labs	23	Experimental	6	Jupyter Notebook
39	aurelio-amerio/GenSBI-examples Examples and benchmarks for the GenSBI library	22	Experimental	—	Python
40	aiden200/Stable_Diffusion_Implementation Implementation of the Stable Diffusion paper: "High-Resolution Image...	22	Experimental	1	Jupyter Notebook
41	rlaneth/figment-fusion A Jupyter notebook for AI image generation	21	Experimental	8	Jupyter Notebook
42	abdur75648/Echo-DND Official implementation of Echo-DND: A Dual Noise Diffusion Model for Robust...	21	Experimental	8	Python
43	hearmeneigh/dataset-rising Toolchain for creating custom datasets and training Stable Diffusion (1.x,...	20	Experimental	18	Python
44	inuwamobarak/img-2-img-generation This repository contains a stable diffusion costumed pipeline for generating...	20	Experimental	7	Jupyter Notebook
45	mradovic38/sprite-flow Flow-based generative model for unguided generation of 128x128 RGBA pixel...	19	Experimental	5	Jupyter Notebook
46	congw729/congw729_log 😉 Log files of reading.	19	Experimental	5	—
47	MrForExample/DC_SSDAE Deep Compression Single-Step Diffusion Autoencoder	18	Experimental	3	Python
48	carlosholivan/AudioGenerationDiffusion State-of-the-art of Audio Generation with Diffusion Models	17	Experimental	5	—
49	Rishikesh-Jadhav/Multi-Modal-Image-Generation-using-Grounding-DINO-SAM-and-Stable-Diffusionv2 A project to combine Grounding-DINO with Meta AI's Segment Anything Model...	17	Experimental	5	Jupyter Notebook
50	rr2203/mnist-diffusion Built a diffusion model on simple handwritten images.	16	Experimental	3	Jupyter Notebook
51	khchan123/stable-diffusion-webui-ec2 A quickstart automation to deploy stable-diffusion-webui with AWS EC2	16	Experimental	4	—
52	hhhaj/IlluminAI This project aims to revolutionize how we understand and manipulate light in...	16	Experimental	3	Python
53	HKUST-LongGroup/Diff-II [CVPR 2025] PyTorch implementation of Diff-II	15	Experimental	25	Python
54	AK2k30/Stable_Diffusion_Model_from_Scratch This project implements a Stable Diffusion model from scratch using PyTorch.	15	Experimental	6	Python
55	johnmartinsson/differentiable-mel-spectrogram The official implementation of DMEL the method presented in the paper "DMEL:...	15	Experimental	22	Python
56	gasparyanartur/neurad-diffusion Master Thesis Project - Diffusion Models are used to enhance Neurad Radiance...	14	Experimental	4	Python
57	mountain/metafont-glyphs A metafont-glyphs dataset which facilitate people to define CJK-like glyphs...	14	Experimental	11	Python
58	klae01/ddim-audio Denoising Diffusion Implicit Models	14	Experimental	12	Python
59	dan-crdll/nn_project_dreamdiffusion Re-implementation of the method proposed in ''DreamDiffusion: Generating...	13	Experimental	8	Jupyter Notebook
60	Shaadalam9/gans-traffic This repository contains the code and analysis for the research paper...	13	Experimental	2	HTML
61	pittisl/FreezeAsGuard Code for paper "FreezeAsGuard: Mitigating Illegal Adaptation of Diffusion...	13	Experimental	5	Python
62	hunter-meloche/sd-diffusion-and-dragons stable-diffusion-web-ui extension that generates detailed text descriptions...	13	Experimental	6	Python
63	Eddycrack864/Stable-Diffusion-with-Civitai-and-Hugging-Face-for-Colab Stable Diffusion Automatic 1111 with support for Hugging Face and Civitai...	13	Experimental	8	Jupyter Notebook
64	AlexandreNap/Denoising-Diffusion-Probabilistic-Model-toy-project Simple Denoising Diffusion Probabilistic Model for 2d data points	13	Experimental	6	Jupyter Notebook
65	teKsafari/blacksnow-cc stateful + deterministic generative engine	13	Experimental	2	—
66	ArchieMeng/stable-diffusion-notebookui Colab notebook for quick Stable Diffusion model evaluations.	12	Experimental	4	Jupyter Notebook
67	YashSharma/AudioDiffusionClassification We use audio diffusion models for improving the classification performance.	12	Experimental	4	Jupyter Notebook
68	cure-lab/DiffGuard [ICCV 2023] The official implementation of paper "DiffGuard: Semantic...	12	Experimental	18	Python
69	ES7/Stable-Diffusion-from-Scratch In this repo I have implemented Stable Diffusion from Scratch using PyTorch.	12	Experimental	3	Python
70	qtrrb/diffusion A fork of stable diffusion that adds an optimizations, features and an API	12	Experimental	3	Python
71	Yuan-ManX/AIGC-Development-Diffusion This is a library of models for artificial intelligence-generated art,...	12	Experimental	3	Python
72	speckly/sucorn_loras Low-Rank Adaptations for Stable Diffusion	11	Experimental	—	—
73	siddthartha/enso-darknet A simple asynchronuous json API for running Stable Diffusion generation...	11	Experimental	2	Rust
74	alexander-koch/tof2cta.c Port of TOF-MRA to CTA image-to-image translation in C	11	Experimental	—	C
75	Utsab-2010/Diffusion-LM Implementing Diffusion on a Small-Language-Model inspired by the Diffusion-LM paper	11	Experimental	—	Jupyter Notebook
76	snowy-the-arctic-fox/Stable-Diffusion A stable diffusion AI just like DALLE, but with a little spice.	10	Experimental	1	Jupyter Notebook
77	RickDYang/mnist-diffusion This repositry is for AI beginers to learn Diffusion model from code scratch.	10	Experimental	1	Python
78	HliasOuzounis/SPVD-Diffusion A repository of my Master's Thesis: Accelerating diffusion models for...	10	Experimental	3	Jupyter Notebook