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 54 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 54 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	spqb/adabmDCApy Pytorch implementation of adabmDCA	49	Emerging	11	Python
3	opendilab/GenerativeRL Python library for solving reinforcement learning (RL) problems using...	46	Emerging	189	Python
4	AstraZeneca/DiffAbXL The official implementation of DiffAbXL benchmarked in the paper "Exploring...	39	Emerging	90	Python
5	G-U-N/UniRL a unified reinforcement learning toolbox for joint RL on language models and...	37	Emerging	76	Python
6	kidist-amde/ddro We introduce the direct document relevance optimization (DDRO) for training...	37	Emerging	36	Python
7	GIST-CSBL/G2D-Diff Official repository of the G2D-Diff	37	Emerging	7	Jupyter Notebook
8	SimoneFaraulo/Attribute-Guided-Face-Generation-A-Mamba-Diffusion-and-VAE-Framework A comparative Generative AI framework for conditional face synthesis....	37	Emerging	2	Python
9	zjowowen/GenerativeRL_Preview Python library for solving reinforcement learning (RL) problems using...	36	Emerging	11	Python
10	eashanadhikarla/DMLLIE Diffusion Models For Low-Light Image Enhancement: A Multi-Perspective...	33	Emerging	5	—
11	rajoy99/keshik Oversample class imbalanced tabular data by Denoising Diffusion...	33	Emerging	5	Python
12	gy65896/Neptune-X [NeurIPS 2025 Spotlight] Neptune-X: Active X-to-Maritime Generation for...	32	Emerging	38	Python
13	spqb/adabmDCA adabmDCA 2.0 – a flexible but easy-to-use package for Direct Coupling Analysis	31	Emerging	8	—
14	HySonLab/DiffPlace DiffPlace: A Conditional Diffusion Framework for Simultaneous VLSI Placement...	31	Emerging	7	Python
15	LarsDoorenbos/generate-spectra Repository for "Galaxy spectroscopy without spectra: Galaxy properties from...	31	Emerging	14	Python
16	bmwas/generativebrainapi A microservice to expose a latent diffusion model for 3D Brain T1-weighted...	30	Emerging	2	Python
17	manonpbart/Machine-Learning-for-Optical-Communication Code for simulating higher order spatial modes in the presence of...	29	Experimental	3	Jupyter Notebook
18	ytakzk/generative_modelling_with_design_constraints Reinforcement learning for furniture generation	28	Experimental	13	JavaScript
19	dutchcyclinglifestyle/dutchcyclinglifestyle Develop a happier and healthier street with our cycling lifestyle.	28	Experimental	10	Jupyter Notebook
20	MhmdEsml/DDPM-Digital-Rock-Analysis Generating synthetic carbonate and sandstone CT scan images by Denoising...	28	Experimental	7	Python
21	novaia/ntg Generative neural networks for 3D terrain.	27	Experimental	32	Python
22	Y-Research-SBU/Ouroboros Official Repository for Ouroboros - ICCV 2025	27	Experimental	20	Python
23	giannisdaras/ambient-omni [NeurIPS 2025, Spotlight]: Ambient-o: Training Good models with Bad Data.	24	Experimental	31	Python
24	beneboeck/phy-inf-gen-mod-wireless Source code of the Paper "Physics-Informed Generative Modeling of Wireless...	24	Experimental	3	Python
25	caioazevedo-mdm/Hybrid_VFM_Slugging Project developed for Flow Assurance in Oil & Gas. Generates synthetic...	23	Experimental	2	Python
26	hhhaj/IlluminAI This project aims to revolutionize how we understand and manipulate light in...	23	Experimental	3	Python
27	ictlab-unict/not-with-my-name This is an official implementation for "Not with my name! Inferring artists'...	22	Experimental	15	Python
28	avitai/artifex A research-focused modular generative modeling library built on JAX/Flax NNX	22	Experimental	1	Python
29	ayushr6/sd-webui-automation Easlisy generate images using a datasets/list of prompts, using the...	22	Experimental	17	Python
30	danijar/fidjax Frechet Inception Distance in JAX	21	Experimental	7	Python
31	DanishjeetSingh/denoising_diffusion_models PyTorch Implementation of Denoising Diffusion Probabilistic Models	21	Experimental	1	Python
32	piedeboer96/ECG-Signal-Denoising BSc Thesis: Conditional Diffusion Models for ECG Signal Denoising (June 2024)	21	Experimental	7	Jupyter Notebook
33	Way-Yuhao/WassDiff A generative precipitation downscaling framework that models climate extremes.	21	Experimental	3	Jupyter Notebook
34	NITHISHM2410/latent-diffusion-tf LDM using tensorflow.	20	Experimental	5	Python
35	yxshee/generativeai diffusion models	20	Experimental	2	Jupyter Notebook
36	darioShar/pytorch-md4 Pytorch Implementation of MD4: Simplified and Generalized Masked Diffusion...	20	Experimental	11	Python
37	YaNgZhAnG-V5/minimalist_concept_erasure [ICML25] Minimalist Concept Erasure in Generative Models	19	Experimental	16	Python
38	adrian-baule/jumpAI Implementation of a generative jump-diffusion model	19	Experimental	—	Wolfram Language
39	6DEADSHOT9/Potato-Diffusion Potato Diffusion is a hands-on repository that demystifies diffusion models...	19	Experimental	1	Python
40	StevenYuan666/Awesome-Diffusion-Models-for-NLP 📰 Must-read papers on Diffusion Models for Text Generation 🔥	19	Experimental	19	—
41	constantbourdrez/Probabilistic_forecasting_DART This repo contains my non-sensitive scripts for my internship at Oxford...	18	Experimental	2	Python
42	metric-space/diffusion-probabilistic-models-JAX Jax + Equinox code reference for Deep Unsupervised Learning using...	18	Experimental	1	Python
43	I-Halder/A-Solvable-Generative-Model-with-a-Linear-One-Step-Denoiser [HiLD@ICML 2025] Memorization to generalization transition in diffusion models	17	Experimental	—	Python
44	Huk-David/Diffusion-and-Flow-based-Copulas This repository contains the code for "Diffusion and Flow-based Copulas:...	14	Experimental	1	Python
45	2stacks/dd-creator dd-creator is a Gradio and Python based diffusion model dataset creator. It...	14	Experimental	1	Python
46	BoltzmannEntropy/DDPM Book: The Art and Science of Diffusion Models in Generative AI	13	Experimental	14	—
47	minjaekimartist/Diffusion-Wavelet-Transformer Diffusion-Wavelet-Transformer(DWT)	13	Experimental	—	Python
48	SvenPfiffner/SkinMachine SkinMachine, a simple UNet diffusion model to generate novel Minecraft skins...	11	Experimental	—	Jupyter Notebook
49	aidan-ferguson/autonomous-driving-gen-ai Researching Generative AI for Autonomous Driving perception simulation	11	Experimental	—	Jupyter Notebook
50	Huk-David/Ratio-Copula This repository contains the code for "Your copula is a classifier in...	11	Experimental	4	Jupyter Notebook
51	lilitpghs/decoding-innovation-dynamics-in-generative-ai Analysis of Generative AI innovation dynamics using patent text mining and...	11	Experimental	—	—
52	Sid7on1/enhanced_csai_250815769v1_scenegen_single_image_3d_scene_generation_in_one_ AI-Generated Project:...	10	Experimental	1	Python
53	afondiel/how-diffusion-models-work-crash-course-DLAI Diffusion Models crash course with Pytorch from DeepLearningAI	10	Experimental	2	Jupyter Notebook
54	taegyeong-lee/Exploring-Diffusion-Models-for-Generative-Forecasting-of-Financial-Charts Exploring Diffusion Models for Generative Forecasting of Financial Charts, arXiv	10	Experimental	1	Python

Comparisons in this category

GenerativeRL and GenerativeRL_Preview (46 vs 36)