PUZZLE: Efficiently Aligning Large Language Models through Light-Weight Context Switch. - Kinman Lei; Yuyang Jin 001; Mingshu Zhai; Kezhao Huang; Haoxing Ye; Jidong Zhai (USENIX ATC, 2024)

FlashTensor: Optimizing Tensor Programs by Leveraging Fine-grained Tensor Property. - Runxin Zhong; Yuyang Jin 001; Chen Zhang 001; Kinman Lei; Shuangyu Li; Jidong Zhai (PPoPP, 2025)

mTuner: Accelerating Parameter-Efficient Fine-Tuning on Multi-GPU Servers with Elastic Tensor. - Kezhao Huang; Siqi Zhu; Mingshu Zhai; Liyan Zheng 001; Kinman Lei; Jiaao He; Yuyang Jin 001; Jidong Zhai (USENIX ATC, 2025)