Ge Zhu (朱舸)'s starred repositories
rectified-flow-pytorch
Implementation of rectified flow and some of its followup research / improvements in Pytorch
disco-diffdock
Code for the paper DisCo-Diff: Enhancing Continuous Diffusion Models with Discrete Latents, ICML 2024
E-LatentLPIPS
Unofficial Implementation of E-LatentLPIPS(Ensembled-LatentLPIPS) of Diffusion2GAN
titok-pytorch
Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"
build-nanogpt
Video+code lecture on building nanoGPT from scratch
schedule_free
Schedule-Free Optimization in PyTorch
vit-pytorch
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
Langevin_Dynamics_Sampling_2D
Langevin_Dynamics_Sampling_2D
Lumina-T2X
Lumina-T2X is a unified framework for Text to Any Modality Generation
imageinwords
Data release for the ImageInWords (IIW) paper.