Diffusion Model Frameworks Generative AI Tools

Foundational frameworks and implementations for building, training, and experimenting with diffusion models across various architectures (DDPM, DDIM, score-based, energy-based). Does NOT include domain-specific applications (image generation, video synthesis, financial forecasting) or inference tools.

There are 84 diffusion model frameworks tools tracked. 1 score above 50 (established tier). The highest-rated is soran-ghaderi/torchebm at 54/100 with 81 stars.

Get all 84 projects as JSON

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

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

#	Tool	Score	Tier	Stars	Language
1	soran-ghaderi/torchebm 🍓 Build and train energy-based and diffusion models in PyTorch ⚡.	54	Established	81	Python
2	opendilab/GenerativeRL Python library for solving reinforcement learning (RL) problems using...	46	Emerging	189	Python
3	spqb/adabmDCApy Pytorch implementation of adabmDCA	42	Emerging	11	Python
4	kidist-amde/ddro We introduce the direct document relevance optimization (DDRO) for training...	41	Emerging	36	Python
5	AstraZeneca/DiffAbXL The official implementation of DiffAbXL benchmarked in the paper "Exploring...	39	Emerging	90	Python
6	G-U-N/UniRL a unified reinforcement learning toolbox for joint RL on language models and...	37	Emerging	76	Python
7	SimoneFaraulo/Attribute-Guided-Face-Generation-A-Mamba-Diffusion-and-VAE-Framework A comparative Generative AI framework for conditional face synthesis....	36	Emerging	2	Python
8	gy65896/Neptune-X [NeurIPS 2025 Spotlight] Neptune-X: Active X-to-Maritime Generation for...	32	Emerging	38	Python
9	eashanadhikarla/DMLLIE Diffusion Models For Low-Light Image Enhancement: A Multi-Perspective...	31	Emerging	5	—
10	GIST-CSBL/G2D-Diff Official repository of the G2D-Diff	30	Emerging	7	Jupyter Notebook
11	spqb/adabmDCA adabmDCA 2.0 – a flexible but easy-to-use package for Direct Coupling Analysis	30	Emerging	8	—
12	rajoy99/keshik Oversample class imbalanced tabular data by Denoising Diffusion...	29	Experimental	5	Python
13	zjowowen/GenerativeRL_Preview Python library for solving reinforcement learning (RL) problems using...	29	Experimental	11	Python
14	novaia/ntg Generative neural networks for 3D terrain.	27	Experimental	32	Python
15	CogitoNTNU/DiffusionModel The application uses a Deep Learning model to generate new car images from...	27	Experimental	6	Jupyter Notebook
16	bodin-e/linear-combinations-of-latents Implementation of LOL from "Linear Combinations of Latents in Generative...	26	Experimental	4	Jupyter Notebook
17	LarsDoorenbos/generate-spectra Repository for "Galaxy spectroscopy without spectra: Galaxy properties from...	26	Experimental	14	Python
18	anujinho/code Code for the paper "Blockwise Control for Denoising Diffusion Models"	24	Experimental	4	Jupyter Notebook
19	HySonLab/DiffPlace DiffPlace: A Conditional Diffusion Framework for Simultaneous VLSI Placement...	24	Experimental	7	Python
20	giannisdaras/ambient-omni [NeurIPS 2025, Spotlight]: Ambient-o: Training Good models with Bad Data.	24	Experimental	31	Python
21	bmwas/generativebrainapi A microservice to expose a latent diffusion model for 3D Brain T1-weighted...	23	Experimental	2	Python
22	rahulkhunte/rahulk-ddpm DDPM (Denoising Diffusion Probabilistic Models) implemented from scratch in...	23	Experimental	1	Python
23	avitai/artifex A research-focused modular generative modeling library built on JAX/Flax NNX	23	Experimental	1	Python
24	chrismado/dit-image-generation From-scratch PyTorch Diffusion Transformer (DiT) for class-conditional image...	22	Experimental	—	Python
25	manonpbart/Machine-Learning-for-Optical-Communication Code for simulating higher order spatial modes in the presence of...	22	Experimental	3	Jupyter Notebook
26	ytakzk/generative_modelling_with_design_constraints Reinforcement learning for furniture generation	21	Experimental	13	JavaScript
27	Y-Research-SBU/Ouroboros Official Repository for Ouroboros - ICCV 2025	21	Experimental	20	Python
28	dutchcyclinglifestyle/dutchcyclinglifestyle Develop a happier and healthier street with our cycling lifestyle.	21	Experimental	10	Jupyter Notebook
29	MhmdEsml/DDPM-Digital-Rock-Analysis Generating synthetic carbonate and sandstone CT scan images by Denoising...	21	Experimental	7	Python
30	ovega14/sun_diffusion Diffusion for SU(N) degrees of freedom	21	Experimental	2	Wolfram Language
31	caioazevedo-mdm/Hybrid_VFM_Slugging Project developed for Flow Assurance in Oil & Gas. Generates synthetic...	21	Experimental	2	Python
32	BASE-Laboratory/PorousDiff Generative Surrogate Modelling for battery electrode microstructures using...	20	Experimental	1	—
33	adrian-baule/jumpAI Implementation of a generative jump-diffusion model	19	Experimental	—	Wolfram Language
34	FLUOLANET/Oxford_102_Flower_Tiny_DDPM 🛴Work in Progress	19	Experimental	—	Jupyter Notebook
35	Aaronaferns/Building-Diffusion-Models Experimental sandbox for diffusion: Focus on minimal architectural changes...	19	Experimental	—	Python
36	beneboeck/phy-inf-gen-mod-wireless Source code of the Paper "Physics-Informed Generative Modeling of Wireless...	18	Experimental	3	Python
37	Tinker-Twins/Sim2Real-Diffusion Sim2Real Diffusion: Learning Cross-Domain Adaptive Representations for...	18	Experimental	4	Jupyter Notebook
38	theodelfolie/MinDiffusion A minimal, educational implementation of a *Denoising Diffusion...	17	Experimental	2	Python
39	danijar/fidjax Frechet Inception Distance in JAX	16	Experimental	7	Python
40	AmzadHossainrafis/Diffusion_model Diffusion models are a class of generative models that excel in image...	16	Experimental	1	Jupyter Notebook
41	zapaishchykova/SynthBrainGrow Repo for the paper "SynthBrainGrow: Synthetic Diffusion Brain Aging for ...	16	Experimental	4	Jupyter Notebook
42	ictlab-unict/not-with-my-name This is an official implementation for "Not with my name! Inferring artists'...	15	Experimental	15	Python
43	CryAndRRich/npmod Building a simple deep learning framework and some machine learning/deep...	15	Experimental	—	Python
44	I-Halder/A-Solvable-Generative-Model-with-a-Linear-One-Step-Denoiser [HiLD@ICML 2025] Memorization to generalization transition in diffusion models	15	Experimental	—	Python
45	ayushr6/sd-webui-automation Easlisy generate images using a datasets/list of prompts, using the...	15	Experimental	17	Python
46	AgamChopra/Diffusion-generator Diffusion based generative model to synthesize 2d images from random noise...	14	Experimental	3	Python
47	Way-Yuhao/WassDiff A generative precipitation downscaling framework that models climate extremes.	14	Experimental	3	Jupyter Notebook
48	DanishjeetSingh/denoising_diffusion_models PyTorch Implementation of Denoising Diffusion Probabilistic Models	14	Experimental	1	Python
49	yhsure/drifting-models-pytorch PyTorch implementation of Generative Modeling via Drifting	14	Experimental	3	TeX
50	egeyavuzcan/diffusion-flow-models-research A comprehensive collection of research papers and resources on...	14	Experimental	3	—
51	Wassely5/Generative-AI-Diffusion-Models A foundational framework and implementation of Denoising Diffusion...	14	Experimental	—	Python
52	mohamed-elkholy95/diffusion-image-gen Diffusion model image generation with U-Net architecture, linear/cosine...	14	Experimental	—	Python
53	piedeboer96/ECG-Signal-Denoising BSc Thesis: Conditional Diffusion Models for ECG Signal Denoising (June 2024)	14	Experimental	7	Jupyter Notebook
54	darioShar/pytorch-md4 Pytorch Implementation of MD4: Simplified and Generalized Masked Diffusion...	13	Experimental	11	Python
55	dineshdaultani/DiffAUD [CVPR 2024] Source code for "Diffusion-Based Adaptation for Classification...	13	Experimental	6	Python
56	NITHISHM2410/latent-diffusion-tf LDM using tensorflow.	13	Experimental	5	Python
57	YaNgZhAnG-V5/minimalist_concept_erasure [ICML25] Minimalist Concept Erasure in Generative Models	13	Experimental	16	Python
58	yxshee/generativeai diffusion models	13	Experimental	2	Jupyter Notebook
59	StevenYuan666/Awesome-Diffusion-Models-for-NLP 📰 Must-read papers on Diffusion Models for Text Generation 🔥	12	Experimental	19	—
60	metric-space/diffusion-probabilistic-models-JAX Jax + Equinox code reference for Deep Unsupervised Learning using...	12	Experimental	1	Python
61	Huk-David/Diffusion-and-Flow-based-Copulas This repository contains the code for "Diffusion and Flow-based Copulas:...	12	Experimental	1	Python
62	6DEADSHOT9/Potato-Diffusion Potato Diffusion is a hands-on repository that demystifies diffusion models...	12	Experimental	1	Python
63	TwelfthBam/flower-diffusion-model A custom implementation of a Denoising Diffusion Probabilistic Model (DDPM)...	12	Experimental	1	Python
64	2stacks/dd-creator dd-creator is a Gradio and Python based diffusion model dataset creator. It...	12	Experimental	1	Python
65	datvodinh/diffusion-model Simple implementation of Diffusion model from scratch with Pytorch Lightning	12	Experimental	4	Python
66	haldersourav/mnist-diffusion A PyTorch implementation of diffusion model on MNIST dataset	12	Experimental	3	Python
67	Radinyn/diffusion-snake Diffusion based Snake world model	12	Experimental	3	Python
68	NITHISHM2410/diffusion-model-tf-ddpm The repository currently encompasses diffusion model implemented using the...	12	Experimental	4	Python
69	NITHISHM2410/diffusion-model-tf-generativefill This repo uses diffusion models to apply generative fill on regions of images	12	Experimental	3	Python
70	alcazar90/ddpo-celebahq Reward Finetuning on Diffusion Models using DDPO	11	Experimental	2	Python
71	danilodjor/denoising-diffusion Implementation of the original denoising diffusion probabilistic model...	11	Experimental	2	Python
72	constantbourdrez/Probabilistic_forecasting_DART This repo contains my non-sensitive scripts for my internship at Oxford...	11	Experimental	2	Python
73	lilitpghs/decoding-innovation-dynamics-in-generative-ai Analysis of Generative AI innovation dynamics using patent text mining and...	11	Experimental	—	—
74	ISCODEVUTB/EcomFruitAI Fruit Images with Diffusion Models	11	Experimental	—	Jupyter Notebook
75	liadsantos/ddpm-mnist Implementation from scratch of DDPM with classifier-free guidance using MNIST dataset	11	Experimental	—	Jupyter Notebook
76	PrudhviGudla/ddpm-ddim-2d-manifold A from scratch PyTorch implementation of DDPM and DDIM samplers trained on...	11	Experimental	—	Jupyter Notebook
77	randomaccess2023/KPC_DDPM_128 This repository hosts the codebase for the conference paper titled...	11	Experimental	—	Jupyter Notebook
78	Play3rZer0/diffusion-models-ai A project that simulates the forward and reverse diffusion process in...	11	Experimental	—	Python
79	minjaekimartist/Diffusion-Wavelet-Transformer Diffusion-Wavelet-Transformer(DWT)	11	Experimental	—	Python
80	Musthafa-D/Image-generation-of-Cifar10-using-Diffusion-Models Image generation on the CIFAR-10 dataset using diffusion models implemented...	11	Experimental	—	Python
81	Musthafa-D/Image-generation-of-Mnist-using-Diffusion-Models MNIST image generation using diffusion models implemented in PyTorch.	11	Experimental	—	Python
82	Musthafa-D/Image-generation-of-CelebA-using-Diffusion-Models CelebA image generation using diffusion models implemented in PyTorch.	11	Experimental	—	Python
83	sustainable-computing/PrivDiffuser Code for "PrivDiffuser: Privacy-Guided Diffusion Model for Data Obfuscation...	11	Experimental	—	Python
84	ES7/Diffusion-Models-UNet Complete guide on Diffusion Model from definition to training and inference	10	Experimental	1	Jupyter Notebook

Comparisons in this category

GenerativeRL and GenerativeRL_Preview (46 vs 29)