Generative Models Python Packages

piq

Measures and metrics for image2image tasks. PyTorch.

cd-fvd

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

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.

deepsmiles

DeepSMILES - A variant of SMILES for use in machine-learning

torchgen

:package: Ready to use implementations of state-of-the-art generative models in PyTorch

zea

zea: A Toolbox for Cognitive Ultrasound Imaging

chuchichaestli

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

sdgym

Benchmarking synthetic data generation methods.

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.

focal-frequency-loss

[ICCV 2021] Focal Frequency Loss for Image Reconstruction and Synthesis

monai-generative

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

raylab

Reinforcement learning algorithms in RLlib

genblaze-core

Genblaze is an open source Python SDK for orchestrating generative AI media pipelines across video, audio, and image providers with built in provenance for every output.

genblaze-s3

Genblaze is an open source Python SDK for orchestrating generative AI media pipelines across video, audio, and image providers with built in provenance for every output.

protaska-gpt

Unleash the Potential of Datasets with Intelligent Tasks, Tutorials, and Algorithm Recommendations.

genblaze-openai

Genblaze is an open source Python SDK for orchestrating generative AI media pipelines across video, audio, and image providers with built in provenance for every output.

genblaze

Genblaze is an open source Python SDK for orchestrating generative AI media pipelines across video, audio, and image providers with built in provenance for every output.

genblaze-gmicloud

Genblaze is an open source Python SDK for orchestrating generative AI media pipelines across video, audio, and image providers with built in provenance for every output.

sbiax

Fast, lightweight and parallelised simulation-based inference in JAX.

genblaze-elevenlabs

Genblaze is an open source Python SDK for orchestrating generative AI media pipelines across video, audio, and image providers with built in provenance for every output.

genblaze-replicate

Genblaze is an open source Python SDK for orchestrating generative AI media pipelines across video, audio, and image providers with built in provenance for every output.

medlatents

Generative modelling of tokenized medical images: autoregressive, MaskGIT, diffusion, flow matching, and Bayesian flow.

genblaze-nvidia

Genblaze is an open source Python SDK for orchestrating generative AI media pipelines across video, audio, and image providers with built in provenance for every output.

genblaze-google

Genblaze is an open source Python SDK for orchestrating generative AI media pipelines across video, audio, and image providers with built in provenance for every output.