I’m a M.S. student in Computer Science at Tsinghua University, advised by Prof. Zhi Wang.
My research focuses on machine learning systems, especially distributed training and inference for large models.
I build systems that make AI faster, more efficient, and more scalable.
Email: luo-jj24@mails.tsinghua.edu.cn
Educations
- Tsinghua University — M.S. in Computer Science, 2024 – Present
- Southern University of Science and Technology — B.E. in Computer Science, 2020 – 2024
Publications
[NeurIPS 2025] CompactFusion: Accelerating Parallel Diffusion Model Serving with Residual Compression
Jiajun Luo*, Yicheng Xiao*, Jianru Xu, Yangxiu You, Rongwei Lu, Chen Tang, Jingyan Jiang, Zhi Wang
📄 Paper · 💻 Code
A parallel framework that mitigates communication bottlenecks in diffusion inference via residual compression, reducing inter-GPU traffic by 8×–16×.
[ICCV 2025] DICE: Staleness-Centric Optimizations for Parallel Diffusion MoE Inference
Jiajun Luo*, Lizhuo Luo*, Jianru Xu, Jiajun Song, Rongwei Lu, Chen Tang, Zhi Wang
📄 Paper · 💻 Code
An asynchronous expert-parallel framework for diffusion MoE that overlaps communication and computation.