MultiPath

Squirrel

PyTorch implementation of Transformer-based Neural Machine Translation

NA-NMT

Non-autoregressive Neural Machine Translation (not a full version)

MetaNMT

Meta learning for Neural Machine Translation

DepthwiseConv2d

Efficient implementation of Depthwise Conv2d. The main code is from https://github.com/MegEngine

splatter-image

Official implementation of `Splatter Image: Ultra-Fast Single-View 3D Reconstruction'

image2edge

image to edge with pidinet

stable-dreamfusion

A working implementation of text-to-3D dreamfusion, powered by stable diffusion.

threestudio

A unified framework for 3D content generation.

tiny-cuda-nn

Lightning fast & tiny C++/CUDA neural network framework

AnimateDiff

Official implementation of AnimateDiff.

cv

Current CV of Jiatao Gu

DiT

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

DragGAN_unofficial

Wild implementation of "Drag Your GAN: Interactive Point-based Manipulation on the Generative Image Manifold"

dreamgaussian

Generative Gaussian Splatting for Efficient 3D Content Creation

gangealing

Official PyTorch Implementation of GAN-Supervised Dense Visual Alignment (CVPR 2022 Oral, Best Paper Finalist)

gaussian-splatting

Original reference implementation of "3D Gaussian Splatting for Real-Time Radiance Field Rendering"

GECO

gpt-fast

Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.

gsgen

Text-to-3D using Gaussian Splatting

imaginaire-stable

NVIDIA's Deep Imagination Team's PyTorch Library

LAION-5B-WatermarkDetection

Lumina-T2X

Lumina-T2X is a unified framework for Text to Any Modality Generation

nerfacc

A General NeRF Acceleration Toolbox in PyTorch.

ngp_pl

Instant-ngp in pytorch+cuda trained with pytorch-lightning (with only few lines of legible code)

OpenLRM

An open-source impl. of Large Reconstruction Models

prompt-to-prompt

Score-Entropy-Discrete-Diffusion

Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)

stablediffusion

High-Resolution Image Synthesis with Latent Diffusion Models

starless

Starless is a CPU black hole raytracer in numpy suitable for both informative diagrams and decent wallpaper material.

U-ViT

A PyTorch implementation of the paper "All are Worth Words: A ViT Backbone for Diffusion Models".