Publications

2026

1. ArXiv

   AMMA: A Multi-Chiplet Memory-Centric Architecture for Low-Latency 1M Context Attention Serving

   Zhongkai Yu, Haotian Ye, Chenyang Zhou, Ohm Rishabh Venkatachalam, Zaifeng Pan, Zhengding Hu, Junsung Kim, Won Woo Ro, Po-An Tsai, Shuyi Pei, Yangwook Kang, and Yufei Ding

   arXiv preprint arXiv:2604.26103, 2026

   arXiv Bib PDF

   @article{yu2026amma,
     title = {AMMA: A Multi-Chiplet Memory-Centric Architecture for Low-Latency 1M Context Attention Serving},
     author = {Yu, Zhongkai and Ye, Haotian and Zhou, Chenyang and Venkatachalam, Ohm Rishabh and Pan, Zaifeng and Hu, Zhengding and Kim, Junsung and Ro, Won Woo and Tsai, Po-An and Pei, Shuyi and Kang, Yangwook and Ding, Yufei},
     journal = {arXiv preprint arXiv:2604.26103},
     year = {2026},
   }

2. ArXiv

   SimSD: Simple Speculative Decoding in Diffusion Language Models

   Junxia Cui^*, Haotian Ye^*, Runchu Tian^*, Hongcan Guo, Jinya Jiang, Haoru Li, Chaojie Ren, Yiming Huang, Kaijie Zhu, Zhongkai Yu, Kun Zhou, and Jingbo Shang

   arXiv preprint arXiv:2606.02544. ^*Equal contribution , 2026

   arXiv Bib PDF

   @article{cui2026simsd,
     title = {SimSD: Simple Speculative Decoding in Diffusion Language Models},
     author = {Cui, Junxia and Ye, Haotian and Tian, Runchu and Guo, Hongcan and Jiang, Jinya and Li, Haoru and Ren, Chaojie and Huang, Yiming and Zhu, Kaijie and Yu, Zhongkai and Zhou, Kun and Shang, Jingbo},
     journal = {arXiv preprint arXiv:2606.02544},
     year = {2026},
   }

3. ArXiv

   ChipBench: A Next-Step Benchmark for Evaluating LLM Performance in AI-Aided Chip Design

   Zhongkai Yu, Chenyang Zhou, Yichen Lin, Hejia Zhang, Haotian Ye, Junxia Cui, Zaifeng Pan, Jishen Zhao, and Yufei Ding

   arXiv preprint arXiv:2601.21448, 2026

   arXiv Bib PDF

   @article{yu2026chipbench,
     title = {ChipBench: A Next-Step Benchmark for Evaluating LLM Performance in AI-Aided Chip Design},
     author = {Yu, Zhongkai and Zhou, Chenyang and Lin, Yichen and Zhang, Hejia and Ye, Haotian and Cui, Junxia and Pan, Zaifeng and Zhao, Jishen and Ding, Yufei},
     journal = {arXiv preprint arXiv:2601.21448},
     year = {2026},
   }