Diffusion Models Python Packages

openvino

OpenVINO™ is an open source toolkit for optimizing and deploying AI inference

openvino-dev

OpenVINO™ is an open source toolkit for optimizing and deploying AI inference

cd-fvd

[CVPR 2024] On the Content Bias in Fréchet Video Distance

vsa

A unified inference and post-training framework for accelerated video generation.

st-attn

A unified inference and post-training framework for accelerated video generation.

deepcache

[CVPR 2024] DeepCache: Accelerating Diffusion Models for Free

remove-ai-watermarks

Remove visible and invisible AI watermarks and provenance metadata from images and video. Python library and CLI for SynthID, C2PA, EXIF, IPTC, XMP, and common generative-AI marks.

avitai-artifex

Generative modeling for JAX/Flax NNX: VAEs, GANs, DDPM/DDIM/score/DiT/latent diffusion, normalizing flows, energy-based, autoregressive and geometric (SE(3), protein, point cloud, mesh) models; image, text, audio, molecular, protein, tabular and time-series modalities; a shared Trainer with callbacks and extensions; evaluation through calibrax.

image-reward

[NeurIPS 2023] ImageReward: Learning and Evaluating Human Preferences for Text-to-image Generation

onediff

OneDiff: An out-of-the-box acceleration library for diffusion models.

aquiles-image

A high-performance, memory-efficient inference server for diffusion models (image/video generation), compatible with the OpenAI client

transformerlab

The open source research environment for AI researchers to seamlessly train, evaluate, and scale models from local hardware to GPU clusters.

fastvideo-kernel

A unified inference and post-training framework for accelerated video generation.

onediffx

OneDiff: An out-of-the-box acceleration library for diffusion models.

chuchichaestli

Where you find all the state-of-the-art cooking utensils (salt, pepper, gradient descent... the usual).

simpletuner

A general fine-tuning kit geared toward image/video/audio diffusion models.

hintauc

Hint-AUC (IEEE TVCG 2026): deterministic region-based hint generation, Hint-AUC evaluation and the DetFill colorization model — code, checkpoints, data and reproduction package

torchebm

🍓 Simulation-free, GPU-first generative modeling in PyTorch ⚡ Composable primitives for scalable, stable training of modern EBMs, diffusion, flow matching, and Schrödinger bridges.

deepinv

DeepInverse: a PyTorch library for solving imaging inverse problems using deep learning

noai-watermark

Remove invisible AI watermarks (SynthID, StableSignature, TreeRing) and strip AI metadata from images. Open-source CLI & Python toolkit.

openvino-nightly

OpenVINO™ is an open source toolkit for optimizing and deploying AI inference

azula

Diffusion models in PyTorch

monai-generative

MONAI Generative Models makes it easy to train, evaluate, and deploy generative models and related applications

rose-opt

🌹 Rose: Range-Of-Slice Equilibration PyTorch optimizer. Stateless optimization through range-normalized gradient updates.