Ziliang Lai
1 indexed paper
Recent (6 mo)
1With code
0Influential cites
0Benchmarked
0Publications per year
126
Top categories
Distributed×1AI×1
Frequent co-authors
Research Timeline
2026
X-Stage: An Overlooked Pipeline Stage for Communication-Computation Overlap in DiT Inference
This paper introduces X-Stage, a software-visible post-issue pipeline stage to improve communication efficiency in distributed diffusion transformer (DiT) inference, leading to significant speedups for DeepGEMM MegaMoE and Ulysses sequence-parallel attention.
Highlighted terms show continued research focus across papers
Papers
cs.DCcs.AIEmpiricalRecentJul 25, 2026
X-Stage: An Overlooked Pipeline Stage for Communication-Computation Overlap in DiT Inference
Jianwen Xian, Zhiyuan Xu, Yuchen Li, Ziliang Lai +7 more
This paper introduces X-Stage, a software-visible post-issue pipeline stage to improve communication efficiency in distributed diffusion transformer (DiT) inference, leading to significant speedups fo…
View →