Generative Model Python Packages

copulas

A library to model multivariate data using copulas.

sdv

Synthetic data generation for tabular data

image-reward

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

zuko

Normalizing flows in PyTorch

nflows

Normalizing flows in PyTorch

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

prdc

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

azula

Diffusion models in PyTorch

genqc

Generative Quantum Circuits

bayesianflow-for-chem

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

causalegm

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

torchgan

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

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.

gpt2lm

A implimentation of GPT2 varient.

scgen

Single cell perturbation prediction

cplx-gmm

PyPI package for complex-valued Gaussian mixture models with a scikit-learn-style API.

mimikit

Music Modeling Kit

modelbusinesscards

Generating documentation of AI model usage to increase visibility and safety

cplx-mfa

PyPI package for complex-valued mixture of factor analyzers with a scikit-learn-style API.

torchflows

Normalizing flows in PyTorch

causaltorch

A PyTorch library for building generative models with causal constraints

world-machine

Investigating the concept and creation of computational world models