Python generative-model

Open-source Python projects categorized as generative-model

Top 23 Python generative-model Projects

  • jukebox

    Code for the paper "Jukebox: A Generative Model for Music"

    Project mention: Open Source Libraries | /r/AudioAI | 2023-10-02

    openai/jukebox: Music Generation

  • denoising-diffusion-pytorch

    Implementation of Denoising Diffusion Probabilistic Model in Pytorch

    Project mention: Commits · lucidrains/denoising-diffusion-pytorch | /r/u_metareflection | 2023-05-24
  • InfluxDB

    Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

  • stylegan2-pytorch

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

    Project mention: Wikipedia No Longer Considers CNET "Generally Reliable" Source After AI Scandal | | 2024-02-29
  • ALAE

    [CVPR2020] Adversarial Latent Autoencoders

  • SDV

    Synthetic data generation for tabular data

    Project mention: Synthetic data generation for tabular data | | 2024-02-27

    Can someone help me understand the licensing of this?

    It was MIT licensed up until 2022 where it was changed to what it is now, where they say that it will become MIT again 4 years after release... but is that from when the license was changed or the first release of the software in GitHub?

  • DreamCraft3D

    [ICLR 2024] Official implementation of DreamCraft3D: Hierarchical 3D Generation with Bootstrapped Diffusion Prior

    Project mention: AI Image Generator Transforms 2D Images into 3D Images | | 2023-11-07
  • giraffe

    This repository contains the code for the CVPR 2021 paper "GIRAFFE: Representing Scenes as Compositional Generative Neural Feature Fields" (by autonomousvision)

  • WorkOS

    The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.

  • RAVE

    Official implementation of the RAVE model: a Realtime Audio Variational autoEncoder

    Project mention: How could you use AI for music inspiration/ideas? | /r/edmproduction | 2023-05-23

    Also, theres new tools like which can be used to create new sounds, i feel like the more granular you work with AI on art, the better can help you out. Magentajs is pretty cool! Im still just playing with it but is easy to use so far.

  • kogpt

    KakaoBrain KoGPT (Korean Generative Pre-trained Transformer) (by kakaobrain)

  • ImageReward

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

    Project mention: Results of finetuning Avalon TRUvision v2 with image scoring | /r/StableDiffusion | 2023-05-17

    I used Image Reward repo to score generated imaged during training and modified loss function to take score into account.

  • Text2LIVE

    Official Pytorch Implementation for "Text2LIVE: Text-Driven Layered Image and Video Editing" (ECCV 2022 Oral)

  • torch-fidelity

    High-fidelity performance metrics for generative models in PyTorch

    Project mention: [D] A better way to compute the Fréchet Inception Distance (FID) | /r/MachineLearning | 2023-04-10

    The Fréchet Inception Distance (FID) is a widespread metric to assess the quality of the distribution of a image generative model (GAN, Stable Diffusion, etc.). The metric is not trivial to implement as one needs to compute the trace of the square root of a matrix. In all PyTorch repositories I have seen that implement the FID (,,, ...), the authors rely on SciPy's sqrtm to compute the square root of the matrix, which is unstable and slow.

  • nflows

    Normalizing flows in PyTorch

  • magvit

    Official JAX implementation of MAGVIT: Masked Generative Video Transformer (by google-research)

    Project mention: Magvit: Masked Generative Video Transformer | | 2023-06-23

    Looks like it's stuck in Google's release review process. Author hopes to publish this week, per


    Lijun-Yu commented Jun 20, 2023

    I hate the delay in the code release due to company review policies... but hopefully it will be out this week during CVPR.

    The initial version will live at (not yet online as of 06/19), written in Jax/Flax. We are also going to release model weights trained on non-proprietary datasets, along with generated samples, so long as they're approved.

    I'm also happy to help with any potential Pytorch reimplementations.


  • MeshDiffusion

    Official implementation of "MeshDiffusion: Score-based Generative 3D Mesh Modeling" (ICLR 2023 Spotlight)

    Project mention: Generating 3D meshes with machine learning | | 2023-03-18

    Check out our MeshDiffusion model that can generate 3D meshes effortlessly. The code is available here:

  • progen

    Official release of the ProGen models

    Project mention: Large language models generate functional protein sequences across families | | 2023-05-13

    I was supposed to be reply to another comment. The GitHub is from 2022:

  • MidiTok

    MIDI / symbolic music tokenizers for Deep Learning models 🎶

  • Papers-in-100-Lines-of-Code

    Implementation of papers in 100 lines of code.

    Project mention: How do I run this code from Papers in 100 lines of code? | /r/NeuralRadianceFields | 2023-09-22

    I wanted to try the some code written by Maxime Vandegar

  • Copulas

    A library to model multivariate data using copulas.

  • evodiff

    Generation of protein sequences and evolutionary alignments via discrete diffusion models

    Project mention: What’s Next in Protein Design? Microsoft Researchers Introduce EvoDiff: A Groundbreaking AI Framework for Sequence-First Protein Engineering | /r/machinelearningnews | 2023-09-19


  • ml-gmpi

    Official PyTorch implementation of GMPI (ECCV 2022, Oral Presentation)

  • synthcity

    A library for generating and evaluating synthetic tabular data for privacy, fairness and data augmentation.

  • CelebV-HQ

    [ECCV 2022] CelebV-HQ: A Large-Scale Video Facial Attributes Dataset

  • SaaSHub

    SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives

NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020). The latest post mention was on 2024-02-29.

Python generative-model related posts


What are some of the best open-source generative-model projects in Python? This list will help you:

Project Stars
1 jukebox 7,427
2 denoising-diffusion-pytorch 6,492
3 stylegan2-pytorch 3,613
4 ALAE 3,492
5 SDV 1,821
6 DreamCraft3D 1,701
7 giraffe 1,223
8 RAVE 1,128
9 kogpt 991
10 ImageReward 870
11 Text2LIVE 848
12 torch-fidelity 841
13 nflows 782
14 magvit 775
15 MeshDiffusion 714
16 progen 538
17 MidiTok 534
18 Papers-in-100-Lines-of-Code 515
19 Copulas 488
20 evodiff 385
21 ml-gmpi 333
22 synthcity 327
23 CelebV-HQ 306
SaaSHub - Software Alternatives and Reviews
SaaSHub helps you find the best software and product alternatives