publications

2026

1. arXiv

   CommBench: Can LLMs Write Correct and Efficient GPU Communication Code?

   Shuang Ma, Yuyi Li, Yihan Zhang, Hezhi Xie, Danyang Chen, Shuyang Ji, Ziming Mao, Cheng Ji, Ansha Prashanth, Wenting Yang, Yiran Wang, Chihan Cui, Pei Yu Lin, Ion Stoica, and Yang Zhou

   arXiv preprint arXiv:2608.04450, 2026

   Bib HTML

   @article{ma2026commbench,
     title = {{CommBench}: Can {LLMs} Write Correct and Efficient {GPU} Communication Code?},
     author = {Ma, Shuang and Li, Yuyi and Zhang, Yihan and Xie, Hezhi and Chen, Danyang and Ji, Shuyang and Mao, Ziming and Ji, Cheng and Prashanth, Ansha and Yang, Wenting and Wang, Yiran and Cui, Chihan and Lin, Pei Yu and Stoica, Ion and Zhou, Yang},
     journal = {arXiv preprint arXiv:2608.04450},
     year = {2026},
     url = {https://arxiv.org/abs/2608.04450},
   }

2. COLM ’26

   Expert-Choice Routing Enables Adaptive Computation in Diffusion Language Models

   Shuibai Zhang^*, Caspian Zhuang^*, Chihan Cui^*, Zhihan Yang, Fred Zhangzhi Peng, Yanxin Zhang, Haoyue Bai, Zack Jia, Yang Zhou, Guanhua Chen, and Ming Liu

   In Conference on Language Modeling (COLM), 2026

   Bib HTML

   @inproceedings{zhang2026ecdlm,
     title = {Expert-Choice Routing Enables Adaptive Computation in Diffusion Language Models},
     author = {Zhang, Shuibai and Zhuang, Caspian and Cui, Chihan and Yang, Zhihan and Peng, Fred Zhangzhi and Zhang, Yanxin and Bai, Haoyue and Jia, Zack and Zhou, Yang and Chen, Guanhua and Liu, Ming},
     booktitle = {Conference on Language Modeling (COLM)},
     year = {2026},
     url = {https://arxiv.org/abs/2604.01622},
   }

3. OSDI ’26

   UCCL-EP: Portable Expert-Parallel Communication

   Ziming Mao, Yihan Zhang, Chihan Cui, Kaichao You, Zhongjie Chen, Zhiying Xu, Scott Shenker, Costin Raiciu, Yang Zhou, and Ion Stoica

   19th USENIX Symposium on Operating Systems Design and Implementation (OSDI), 2026

   Bib HTML

   @article{mao2025ucclep,
     title = {UCCL-EP: Portable Expert-Parallel Communication},
     author = {Mao, Ziming and Zhang, Yihan and Cui, Chihan and You, Kaichao and Chen, Zhongjie and Xu, Zhiying and Shenker, Scott and Raiciu, Costin and Zhou, Yang and Stoica, Ion},
     journal = {19th USENIX Symposium on Operating Systems Design and Implementation (OSDI)},
     year = {2026},
     url = {https://arxiv.org/abs/2512.19849},
   }

4. SoCC ’26

   Making Sense of DPU Performance for Cloud Data Processing

   Jiasheng Hu, Chihan Cui, Yuanfan Chen, Philip A. Bernstein, Jialin Li, and Qizhen Zhang

   In ACM Symposium on Cloud Computing (SoCC), 2026

   Bib HTML

   @inproceedings{hu2025dpbento,
     title = {Making Sense of {DPU} Performance for Cloud Data Processing},
     author = {Hu, Jiasheng and Cui, Chihan and Chen, Yuanfan and Bernstein, Philip A. and Li, Jialin and Zhang, Qizhen},
     booktitle = {ACM Symposium on Cloud Computing (SoCC)},
     year = {2026},
     url = {https://arxiv.org/abs/2504.05536},
   }