Generative Model Python Packages

copulas

A library to model multivariate data using copulas.

sdv

Synthetic data generation for tabular data

zuko

Normalizing flows in PyTorch

image-reward

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

nflows

Normalizing flows in PyTorch

prdc

Code base for the precision, recall, density, and coverage metrics for generative models. ICML 2020.

denoising-diffusion-pytorch

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

miditok

MIDI / symbolic music tokenizers for Deep Learning models 🎶

stylegan2-pytorch

Simplest working implementation of Stylegan2, state of the art generative adversarial network, in Pytorch. Enabling everyone to experience disentanglement

azula

Diffusion models in PyTorch

scgen

Single cell perturbation prediction

bayesianflow-for-chem

ChemBFN: Bayesian Flow Network Framework for Chemistry Tasks. Developed in Hiroshima University.

molgen

Lightweight toolkit for de novo molecular generation: SMILES & SELFIES tokenizers, CharRNN / MolGPT / VAE models, training, sampling, and MOSES-style metrics.

gpt2lm

A implimentation of GPT2 varient.

scyan

Biology-driven deep generative model for cell-type annotation in cytometry. Scyan is an interpretable model that also corrects batch-effect and can be used for debarcoding or population discovery.

causalegm

CausalEGM: an encoding generative modeling approach to dimension reduction and covariate adjustment in causal inference with observational studies

modelbusinesscards

Model Business Logic Cards to support responsible AI/LLM usage

torchgan

Research Framework for easy and efficient training of GANs based on Pytorch

texturize

🤖🖌️ Generate new photo-realistic textures similar to a source image.

regvelo

inferring regulatory cellular dynamics

evodiff

Generation of protein sequences and evolutionary alignments via discrete diffusion models

genqc

Generative Quantum Circuits

torchflows

Modern normalizing flows in Python. Simple to use and easily extensible.

causaltorch

CausalTorch is a PyTorch library for building generative models with explicit causal constraints. It integrates graph-based causal reasoning with deep learning to create AI systems that respect logical causal relationships.