Generative Model Python Packages

copulas

A library to model multivariate data using copulas.

zuko

Normalizing flows in PyTorch

nflows

Normalizing flows in PyTorch

sdv

Synthetic data generation for tabular data

miditok

MIDI / symbolic music tokenizers for Deep Learning models 🎶

denoising-diffusion-pytorch

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

image-reward

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

lzgraphs

Alignment-free graph models of TCR and BCR repertoires built from Lempel-Ziv and FlashBack string decompositions. Compute sequence generation probability (Pgen) and repertoire diversity, and generate new sequences. Python with a C core.

azula

Diffusion models in PyTorch

prdc

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

bayesianflow-for-chem

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

stylegan2-pytorch

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

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

A General Causal Inference Framework by Encoding Generative Modeling

tabforge-ml

🔨 [NeurIPS 2026] Official implementation of “TabFORGE: A Tabular Foundation Model for Generative Modelling”

scgen

Single cell perturbation prediction

torchgan

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

genqc

Generative Quantum Circuits

mimikit

Music Modeling Kit

evodiff

Generation of protein sequences and evolutionary alignments via discrete diffusion models

molecule-generation

Implementation of MoLeR: a generative model of molecular graphs which supports scaffold-constrained generation

gpt2lm

A implimentation of GPT2 varient.

texturize

🤖🖌️ Generate photo-realistic textures based on source images or (soon) PBR materials. Remix, remake, mashup! Useful if you want to create variations on a theme or elaborate on an existing texture.

lamda-ssl

30 Semi-Supervised Learning Algorithms