PetS: A Unified Framework for Parameter-Efficient Transformers Serving. - Zhe Zhou 002; Xuechao Wei; Jiejing Zhang; Guangyu Sun 003 (USENIX ATC, 2022)

ChatPLUG: Open-Domain Generative Dialogue System with Internet-Augmented Instruction Tuning for Digital Human. - Junfeng Tian; Hehong Chen; Guohai Xu; Ming Yan 008; Xing Gao; Jianhai Zhang; Chenliang Li; Jiayi Liu 004; Wenshen Xu; Haiyang Xu 001; Qi Qian 001; Wei Wang 225; et al. (CoRR, 2023)

RadiK: Scalable and Optimized GPU-Parallel Radix Top-K Selection. - Yifei Li; Bole Zhou; Jiejing Zhang; Xuechao Wei; Yinghan Li; Yingda Chen (ICS, 2024)

POSTER: RadiK: Scalable Radix Top-K Selection on GPUs. - Yifei Li; Bole Zhou; Jiejing Zhang; Xuechao Wei; Yinghan Li; Yingda Chen (PPoPP, 2024)

Static Batching of Irregular Workloads on GPUs: Framework and Application to Efficient MoE Model Inference. - Yinghan Li; Yifei Li; Jiejing Zhang; Bujiao Chen; Xiaotong Chen; Lian Duan; Yejun Jin; Zheng Li; Xuanyu Liu; Haoyu Wang; Wente Wang; Yajie Wang; et al. (CoRR, 2025)