Kai Jiang

PhD student, Tsinghua University

Joined

January 2025

Names

Kai Jiang (Preferred)

Emails

****@smail.nju.edu.cn (Confirmed)

,

****@gmail.com (Confirmed)

Personal Links

Career & Education History

PhD student

Tsinghua University (mails.tsinghua.edu.cn)

2025 – Present

Undergrad student

Nanjing University (smail.nju.edu.cn)

2021 – 2025

Advisors, Relations & Conflicts

PhD Advisor

2025 – Present

Expertise

No areas of expertise listed

Publications

SLA: Beyond Sparsity in Diffusion Transformers via Fine-Tunable Sparse–Linear Attention
Jintao Zhang, Haoxu Wang, Kai Jiang, Shuo Yang, Kaiwen Zheng, Haocheng Xi, Ziteng Wang, Hongzhou Zhu, Min Zhao, Ion Stoica, Joseph E. Gonzalez, Jianfei Chen, Jun Zhu
- ICLR 2026 Poster
- Readers: Everyone
DiffusionNFT: Online Diffusion Reinforcement with Forward Process
Kaiwen Zheng, Huayu Chen, Haotian Ye, Haoxiang Wang, Qinsheng Zhang, Kai Jiang, Hang Su, Stefano Ermon, Jun Zhu, Ming-Yu Liu
- ICLR 2026 Oral
- Readers: Everyone
SageAttention3: Microscaling FP4 Attention for Inference and An Exploration of 8-Bit Training
Jintao Zhang, Jia wei, Haoxu Wang, Pengle Zhang, Xiaoming Xu, Haofeng Huang, Kai Jiang, Jianfei Chen, Jun Zhu
- NeurIPS 2025 spotlight
- Readers: Everyone
Visual Generation Without Guidance
Huayu Chen, Kai Jiang, Kaiwen Zheng, Jianfei Chen, Hang Su, Jun Zhu
- ICML 2025 poster
- Readers: Everyone
Visual Generation Without Guidance
Huayu Chen, Kai Jiang, Kaiwen Zheng, Jianfei Chen, Hang Su, Jun Zhu
- CoRR 2025
- Readers: Everyone

Co-Authors