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

st-attn

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

vsa

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

transformerlab

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

deepcache

[CVPR 2024] DeepCache: Accelerating Diffusion Models for Free

image-reward

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

cd-fvd

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

monai-generative

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

transformerlab-cli

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

deepinv

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

simpletuner

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

aquiles-image

A high-performance, memory-efficient inference server for diffusion models, compatible with the OpenAI client

rose-opt

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

fastvideo-kernel

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

azula

Diffusion models in PyTorch

onediff

an out-of-the-box acceleration library for diffusion models

openvino-nightly

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

avitai-artifex

A research-focused modular generative modeling library built on JAX/Flax NNX

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).

genqc

Generative Quantum Circuits

remove-ai-watermarks

Remove visible and invisible AI watermarks from images (Gemini / Nano Banana, ChatGPT, Stable Diffusion)

diffusion-model-framework

A simple framework for training diffusion models.