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: Mongolian Gabba Goat Techno | | 2023-02-02

    That already exists

  • stylegan2-pytorch

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

    Project mention: Gdy ktoś cię nazywa mnie ruskim agentem, to zawsze przypominam sobie te obrazki. | | 2023-02-05

    Jeszcze ta gęba wzięta z to taka wisienka na torcie

  • Sonar

    Write Clean Python Code. Always.. Sonar helps you commit clean code every time. With over 225 unique rules to find Python bugs, code smells & vulnerabilities, Sonar finds the issues while you focus on the work.

  • ALAE

    [CVPR2020] Adversarial Latent Autoencoders

  • denoising-diffusion-pytorch

    Implementation of Denoising Diffusion Probabilistic Model in Pytorch

    Project mention: Implementation of Google's MusicLM in PyTorch | | 2023-01-30

    Generally it's without weights, but MusicLM is also a WIP more mature implementations have descriptions on how to train them and follow ups on small scale/crowd-sourced experiments & research[1].


  • giraffe

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

  • kogpt

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

  • nflows

    Normalizing flows in PyTorch

  • InfluxDB

    Build time-series-based applications quickly and at scale.. InfluxDB is the Time Series Platform where developers build real-time applications for analytics, IoT and cloud-native services. Easy to start, it is available in the cloud or on-premises.

  • RAVE

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

    Project mention: Ask HN: What weird technical scene are you fond/part of? | | 2022-11-21

    I'm in the deep learning music scene, which is due for its stable diffusion moment in the next year or two. The (primarily) timbre transfer system called RAVE is where I'm starting, and my contribution is to optimize the system to improve training time.


  • torch-fidelity

    High-fidelity performance metrics for generative models in PyTorch

  • Text2LIVE

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

    Project mention: The new neural network from NVIDIA can apply special effects to video using simple text commands. | | 2023-01-31

    The source code of the neural network can be found on GitHub:

  • ml-gmpi

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

    Project mention: Who is building StableDiffusion/DALL-E but for 3D assets? | | 2022-09-07
  • MidiTok

    Convenient MIDI / symbolic music tokenizers for Deep Learning networks 🎶

  • CelebV-HQ

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

    Project mention: CelebV-HQ is a large-scale video facial attributes dataset. It contains 35,666 video clips involving 15,653 identities and 83 manually labeled facial attributes covering appearance, action, and emotion. (by Zhu et al.) | | 2022-11-20


  • DeceiveD

    [NeurIPS 2021] Deceive D: Adaptive Pseudo Augmentation for GAN Training with Limited Data

  • progen

    Official release of the ProGen models

    Project mention: Salesforce/progen: projects and models for protein engineering and design | | 2023-01-29
  • artbench

    Benchmarking Generative Models with Artworks

    Project mention: Public U. Art Club [ so far ✌️ ] | | 2023-01-05

    This example is a data set that's been created for training AI models on differentiating between different styles of art. But at the end of that data set an AI is no more capable of defining what impressionism is than it was before but it can produce new art that appears to be impressionist, surrealist, cubist, whatever based on the examples it's been given and the only reason they don't seem to be copies of any handful of paintings is because the AI has scanned and indexed 1,000 distinct paintings in that style. But a human brain can look at a handful of impressionist painting and intuit the commonalities that make "impressionism" a unique style.

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

    Implementation of papers in 100 lines of code.

  • feed_forward_vqgan_clip

    Feed forward VQGAN-CLIP model, where the goal is to eliminate the need for optimizing the latent space of VQGAN for each input prompt

    Project mention: [D] Hosting AI Art Generative ML Model | | 2022-03-23

    WOMBO I suspect uses the feed forward inferential approach to VQGAN + CLIP (instead of finetuning, predict the final z latent vector for a given text input) which is why their outputs are less sophisticated: as a result there are many deployment optimizations you can do to speed that up, which may be complicated.

  • minimal-text-diffusion

    A minimal implementation of diffusion models for text generation

    Project mention: [D] would diffusion language models make sense? | | 2022-10-24

    In case you’re interested, I have a minimal implementation here:

  • molecule-generation

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

    Project mention: Microsoft AI Researchers Develop MoLeR: A Deep Learning-Based Generative Model That Enables Efficient Drug Design | | 2022-04-30


  • DiffSBDD

    A Euclidean diffusion model for structure-based drug design.

    Project mention: Learn diffusion models with Hugging Face course 🧨 | | 2022-11-04

    I'm referring to recent works like DiffLinker ( and DiffSBDD ( which use diffusion to generate new ligands or fragments of molecules. If there's enough interest, we could find space to include these exciting topics in the course :)

  • stylegan2-flax-tpu

    🖼 Training StyleGAN2 on TPUs in JAX

    Project mention: This Food Does Not Exist | | 2022-10-27
  • timbre_painting

    Hierarchical fast and high-fidelity audio generation

  • 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 2023-02-05.

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 5,772
2 stylegan2-pytorch 3,379
3 ALAE 3,376
4 denoising-diffusion-pytorch 3,133
5 giraffe 1,155
6 kogpt 817
7 nflows 660
8 RAVE 658
9 torch-fidelity 550
10 Text2LIVE 464
11 ml-gmpi 293
12 MidiTok 275
13 CelebV-HQ 240
14 DeceiveD 238
15 progen 217
16 artbench 168
17 Papers-in-100-Lines-of-Code 139
18 feed_forward_vqgan_clip 135
19 minimal-text-diffusion 133
20 molecule-generation 132
21 DiffSBDD 131
22 stylegan2-flax-tpu 127
23 timbre_painting 75
SaaSHub - Software Alternatives and Reviews
SaaSHub helps you find the best software and product alternatives