PyRank
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About

Diffusion Models Python Packages

Python packages with the GitHub topic diffusion-models. Sorted by relevance, with stars and monthly downloads.
openvinotoolkit
openvino

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

1.4M 10K 3K
openvinotoolkit
openvino-dev

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

183K 10K 3K
hao-ai-lab
st-attn

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

148K 3K 329
hao-ai-lab
vsa

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

143K 3K 329
transformerlab
transformerlab

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

126K 5K 511
horseee
deepcache

[CVPR 2024] DeepCache: Accelerating Diffusion Models for Free

54K 967 52
THUDM
image-reward

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

28K 2K 92
songweige
cd-fvd

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

20K 147 7
Project-MONAI
monai-generative

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

19K 758 109
transformerlab
transformerlab-cli

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

13K 5K 511
deepinv
deepinv

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

7K 735 169
bghira
simpletuner

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

6K 3K 280
Aquiles-ai
aquiles-image

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

5K 20 0
MatthewK78
rose-opt

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

4K 60 4
hao-ai-lab
fastvideo-kernel

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

4K 3K 329
probabilists
azula

Diffusion models in PyTorch

3K 131 11
siliconflow
onediff

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

3K 2K 129
openvinotoolkit
openvino-nightly

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

3K 10K 3K
avitai
avitai-artifex

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

3K 1 0
siliconflow
onediffx

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

3K 2K 129
CAIIVS
chuchichaestli

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

2K 5 1
FlorianFuerrutter
genqc

Generative Quantum Circuits

2K 61 15
wiltodelta
remove-ai-watermarks

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

2K 20 2
sachahu1
diffusion-model-framework

A simple framework for training diffusion models.

2K 1 0
    • Data from PyPI, GitHub, ClickHouse, and BigQuery