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.