Yuanzhi Zhu

DiffPIR. "Denoising Diffusion Models for Plug-and-Play Image Restoration", Yuanzhi Zhu, Kai Zhang, Jingyun Liang, Jiezhang Cao, Bihan Wen, Radu Timofte, Luc Van Gool.

prolific_dreamer2d. Unofficial implementation of 2D ProlificDreamer

mini_edm. Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist

OFTSR. [ICLR2026] "OFTSR: One-Step Flow for Image Super-Resolution with Tunable Fidelity-Realism Trade-offs"

SlimFlow. [ECCV2024] "SlimFlow: Training Smaller One-Step Diffusion Models with Rectified Flow", Yuanzhi Zhu, Xingchao Liu, Qiang Liu

DiMO. [ICCV2025] "Di[M]O: Distilling Masked Diffusion Models into One-step Generator", Yuanzhi Zhu, Xi Wang, Stéphane Lathuilière, Vicky Kalogeiton

CRD. Python

ETH_PAI_MindMap. Mind map for ETH course Probabilistic Artificial Intelligence(PAI)

demo_sr_train. A lite version of KAIR with SISR codes

Langevin_Dynamics_Sampling_2D. Langevin_Dynamics_Sampling_2D

SoftDiMO. [ICLR 2026] "Soft-Di[𝙼]O: Improving One-step Discrete Image Generation with Soft Embeddings"

tiny-diffusion. A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.

sjc. Score Jacobian Chaining: Lifting Pretrained 2D Diffusion Models for 3D Generation

CSNLP-Project-ETH. This is the repo for our computational semantics natural language processing course project

yuanzhi-zhu.github.io. Personal Blog