stylegan2-pytorch. Implementation of Analyzing and Improving the Image Quality of StyleGAN (StyleGAN 2) in PyTorch
2.8kvq-vae-2-pytorch. Implementation of Generating Diverse High-Fidelity Images with VQ-VAE-2 in PyTorch
1.8kstyle-based-gan-pytorch. Implementation A Style-Based Generator Architecture for Generative Adversarial Networks in PyTorch
1.1kglow-pytorch. PyTorch implementation of Glow
543alias-free-gan-pytorch. Unofficial implementation of Alias-Free Generative Adversarial Networks. (https://arxiv.org/abs/2106.12423) in PyTorch
504denoising-diffusion-pytorch. Implementation of Denoising Diffusion Probabilistic Models in PyTorch
405ml-papers. My collection of machine learning papers
300swapping-autoencoder-pytorch. Unofficial implementation of Swapping Autoencoder for Deep Image Manipulation (https://arxiv.org/abs/2007.00653) in PyTorch
257mac-network-pytorch. Memory, Attention and Composition (MAC) Network for CLEVR implemented in PyTorch
85vision-transformers-pytorch. Implementation of various Vision Transformers I found interesting
84adaptive-softmax-pytorch. Adaptive Softmax implementation for PyTorch
81sagan-pytorch. Self-Attention Generative Adversarial Networks Implementation in PyTorch
76igebm-pytorch. Implicit Generation and Generalization in Energy Based Models in PyTorch
68depthwise-conv-pytorch. Faster depthwise convolutions for PyTorch
66progressive-gan-pytorch. Implemetatin of Progressive Growing of GANs in PyTorch
64relation-networks-pytorch. Relation Networks for CLEVR implemented in PyTorch
61ocr-pytorch. Object-Contextual Representations for Semantic Segmentation in PyTorch
61imputer-pytorch. Implementation of Imputer: Sequence Modelling via Imputation and Dynamic Programming in PyTorch
58fcos-pytorch. Re-implementation of FCOS for personal study
51knotter. Implementation of Mapper algorithm for Topological Data Analysis
45semantic-pyramid-pytorch. Implementation of Semantic Pyramid for Image Generation (https://arxiv.org/abs/2003.06221) in PyTorch
39id-gan-pytorch. Information Distillation Generative Adversrial Network in PyTorch
27nerf-pytorch. Python
21tensorfn. Weakly opinionated library for implementing ML models. Less boilerplate, More rigor
21taming-transformers-pytorch. Implementation of Taming Transformers for High-Resolution Image Synthesis (https://arxiv.org/abs/2012.09841) in PyTorch
16film-pytorch. Just another implementation of FiLM in PyTorch
14halite. Acceleration framework for Human Alignment Learning
13instant-ngp-pytorch. Study for Instant neural graphics primitives (Unofficial)
11melgan-pytorch. MelGAN and Tacotron 2 in PyTorch
11zeropipe. Experiments with Pipeline Parallelism
8meshfn. Framework for Human Alignment Learning
7nansy-pytorch. Unofficial implementation of Neural Analysis and Synthesis
7sarigan-pytorch. Unofficial implementation of Learning Semantic-aware Normalization for Generative Adversarial Networks (SariGAN) in PyTorch
7arxiv-sanity. arXiv feed tool that heavily inspired by Arxiv Sanity Preserver
6lvpga-pytorch. Implementation of Perceptual Generative Autoencoders in PyTorch
5esrgan-pytorch. ESRGAN: Enhanced Super-Resolution Generative Adversarial Networks in PyTorch
3VoVNet-Detectron. Detectron with VoVNet(CVPRW'19) backbone networks
2DeepSpeed. DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
2sujip. Non-opionated utility library for PyTorch
2semantic-segmentation. Nvidia Semantic Segmentation monorepo
2maskrcnn-pytorch. Re-implementation of Mask R-CNN for personal study
2text-to-text-transfer-transformer. Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"
2small-logan-pytorch. Small-GAN and LOGAN in PyTorch
2inductive-moment-matching. Attempt to replicate Inductive Moment Matching (https://arxiv.org/abs/2503.07565)
2dockerfiles. dockerfiles
2nanodo. Python
1Megatron-DeepSpeed. Ongoing research training transformer language models at scale, including: BERT & GPT-2
1rosinality.github.io. HTML
1alias-free-vae. Experiments on whether reducing aliasing in VAEs or image tokenizers is beneficial
1langfn. A DSL for LLMs
1pygoreadability. Python bindings for go-readability (https://github.com/go-shiori/go-readability)
1cpython. The Python programming language
1synapticmap. Synaptic Map - Simple mindmapping program with directional connections
1malmo-challenge. Task and example code for the Malmo Collaborative AI Challenge
1stylegan2-ada. StyleGAN2 with adaptive discriminator augmentation (ADA) - Official TensorFlow implementation
1usrnet-pytorch. Reimplementation of Deep Unfolding Network for Image Super-Resolution for self study.
1