Huangjie Zheng 郑煌杰

Research Scientist · Apple Machine Learning Research

Cupertino, CA

Portrait of Huangjie Zheng

I am a Research Scientist at Apple Machine Learning Research, led by Samy Bengio. My research spans generative modeling and large language models: diffusion and flow-based models and their distillation into fast one-step generators, diffusion language models and efficient LLM decoding, and self-improvement of LLMs for code generation and reasoning. More broadly, I am interested in making generative models efficient, scalable and principled, drawing on probabilistic and Bayesian methods.

Before joining Apple, I received my PhD in Statistics and Data Sciences from the University of Texas at Austin, advised by Mingyuan Zhou. During my PhD I was a research intern at Microsoft Research, working with Pengcheng He, Weizhu Chen and Chunyuan Li, and at ByteDance AML, working with Jianbo Yuan, Quanzeng You, Guanghan Ning and Hongxia Yang.

I obtained my bachelor’s and master’s degrees from Shanghai Jiao Tong University, advised by Ya Zhang, through a Sino-French program that also took me to Télécom Paris. I was also a research intern at Johns Hopkins University with Lingxi Xie and Alan Yuille.

News

2026-10
Released RSI-Inference Bench, a long-horizon benchmark for self-improving inference systems.
2026-10
New preprint: LoopCD decodes looped transformers better, training-free.
2026-04
Released SSD, code and checkpoints for Embarrassingly Simple Self-Distillation.
2026-01
Three papers accepted at ICLR 2026: CADD, DiffuCoder and WINO.
2025-09
Four papers accepted at NeurIPS 2025, including STARFlow (spotlight) and TADA.
2025-05
TarFlow accepted at ICML 2025 as an oral presentation.

Selected publications Full list →

ICLR 2026

Continuously Augmented Discrete Diffusion Model for Categorical Generative Modeling

Huangjie Zheng, Shansan Gong, Ruixiang Zhang, Tianrong Chen, Jiatao Gu, Mingyuan Zhou, Navdeep Jaitly, Yizhe Zhang

International Conference on Learning Representations (ICLR), 2026

ICLR 2026

DiffuCoder: Understanding and Improving Masked Diffusion Models for Code Generation

Shansan Gong, Ruixiang Zhang, Huangjie Zheng, Jiatao Gu, Navdeep Jaitly, Lingpeng Kong, Yizhe Zhang

International Conference on Learning Representations (ICLR), 2026

NeurIPS 2025 Spotlight

STARFlow: Scaling Latent Normalizing Flows for High-resolution Image Synthesis

Jiatao Gu, Tianrong Chen, David Berthelot, Huangjie Zheng, Yuyang Wang, Ruixiang Zhang, Laurent Dinh, Miguel Angel Bautista, Josh Susskind, Shuangfei Zhai

Advances in Neural Information Processing Systems (NeurIPS) (Spotlight), 2025

ICML 2025 Oral

Normalizing Flows are Capable Generative Models

Shuangfei Zhai, Ruixiang Zhang, Preetum Nakkiran, David Berthelot, Jiatao Gu, Huangjie Zheng, Tianrong Chen, Miguel Angel Bautista, Navdeep Jaitly, Josh Susskind

International Conference on Machine Learning (ICML) (Oral), 2025

ICML 2024

Score Identity Distillation: Exponentially Fast Distillation of Pretrained Diffusion Models for One-Step Generation

Mingyuan Zhou, Huangjie Zheng, Zhendong Wang, Mingzhang Yin, Hai Huang

International Conference on Machine Learning (ICML), 2024

ICLR 2024

Learning Stackable and Skippable LEGO Bricks for Efficient, Reconfigurable, and Variable-Resolution Diffusion Modeling

Huangjie Zheng, Zhendong Wang, Jianbo Yuan, Guanghan Ning, Pengcheng He, Quanzeng You, Hongxia Yang, Mingyuan Zhou

International Conference on Learning Representations (ICLR), 2024

ICLR 2023

Truncated Diffusion Probabilistic Models and Diffusion-based Adversarial Auto-Encoders

Huangjie Zheng, Pengcheng He, Weizhu Chen, Mingyuan Zhou

International Conference on Learning Representations (ICLR), 2023

ICLR 2023

Diffusion-GAN: Training GANs with Diffusion

Zhendong Wang, Huangjie Zheng, Pengcheng He, Weizhu Chen, Mingyuan Zhou

International Conference on Learning Representations (ICLR), 2023