AutoChunk: Automated Activation Chunk for Memory-Efficient Deep Learning Inference. - Xuanlei Zhao; Shenggan Cheng; Guangyang Lu; Haotian Zhou; Bin Jia; Yang You 001 (ICLR, 2024)

HeteGen: Efficient Heterogeneous Parallel Inference for Large Language Models on Resource-Constrained Devices. - Xuanlei Zhao; Bin Jia; Haotian Zhou; Ziming Liu; Shenggan Cheng; Yang You 001 (MLSys, 2024)

FastFold: Optimizing AlphaFold Training and Inference on GPU Clusters. - Shenggan Cheng; Xuanlei Zhao; Guangyang Lu; Jiarui Fang; Tian Zheng; Ruidong Wu; Xiwen Zhang; Jian Peng 001; Yang You 001 (PPoPP, 2024)

AutoChunk: Automated Activation Chunk for Memory-Efficient Long Sequence Inference. - Xuanlei Zhao; Shenggan Cheng; Guangyang Lu; Jiarui Fang; Haotian Zhou; Bin Jia; Ziming Liu; Yang You 001 (CoRR, 2024)

HeteGen: Heterogeneous Parallel Inference for Large Language Models on Resource-Constrained Devices. - Xuanlei Zhao; Bin Jia; Haotian Zhou; Ziming Liu; Shenggan Cheng; Yang You 001 (CoRR, 2024)

DSP: Dynamic Sequence Parallelism for Multi-Dimensional Transformers. - Xuanlei Zhao; Shenggan Cheng; Zangwei Zheng; Zheming Yang; Ziming Liu; Yang You 001 (CoRR, 2024)

WallFacer: Guiding Transformer Model Training Out of the Long-Context Dark Forest with N-body Problem. - Ziming Liu; Shaoyu Wang; Shenggan Cheng; Zhongkai Zhao; Xuanlei Zhao; James Demmel; Yang You 001 (CoRR, 2024)

Faster Vision Mamba is Rebuilt in Minutes via Merged Token Re-training. - Mingjia Shi; Yuhao Zhou 004; Ruiji Yu; Zekai Li; Zhiyuan Liang; Xuanlei Zhao; Xiaojiang Peng; Tanmay Rajpurohit; Shanmukha Ramakrishna Vedantam; Wangbo Zhao; Kai Wang 036; Yang You 001 (CoRR, 2024)

Concerto: Automatic Communication Optimization and Scheduling for Large-Scale Deep Learning. - Shenggan Cheng; Shengjie Lin; Lansong Diao; Hao Wu 077; Siyu Wang; Chang Si; Ziming Liu; Xuanlei Zhao; Jiangsu Du; Wei Lin 016; Yang You 001 (ASPLOS (1), 2025)

Real-Time Video Generation with Pyramid Attention Broadcast. - Xuanlei Zhao; Xiaolong Jin; Kai Wang 036; Yang You 001 (ICLR, 2025)

CDIO: Cross-Domain Inference Optimization with Resource Preference Prediction for Edge-Cloud Collaboration. - Zheming Yang; Wen Ji; Qi Guo; Dieli Hu 001; Chang Zhao; Xiaowei Li; Xuanlei Zhao; Yi Zhao; Chaoyu Gong; Yang You 001 (CoRR, 2025)

Enhance-A-Video: Better Generated Video for Free. - Yang Luo; Xuanlei Zhao; Mengzhao Chen; Kaipeng Zhang; Wenqi Shao; Kai Wang 036; Zhangyang Wang; Yang You 001 (CoRR, 2025)

Dynamic Vision Mamba. - Mengxuan Wu; Zekai Li; Zhiyuan Liang; Moyang Li; Xuanlei Zhao; Samir Khaki; Zheng Zhu; Xiaojiang Peng; Konstantinos N. Plataniotis; Kai Wang 036; Wangbo Zhao; Yang You 001 (CoRR, 2025)

DD-Ranking: Rethinking the Evaluation of Dataset Distillation. - Zekai Li; Xinhao Zhong; Samir Khaki; Zhiyuan Liang; Yuhao Zhou 004; Mingjia Shi; Ziqiao Wang; Xuanlei Zhao; Wangbo Zhao; Ziheng Qin; Mengxuan Wu; Pengfei Zhou; et al. (CoRR, 2025)

REPA Works Until It Doesn't: Early-Stopped, Holistic Alignment Supercharges Diffusion Training. - Ziqiao Wang; Wangbo Zhao; Yuhao Zhou 004; Zekai Li; Zhiyuan Liang; Mingjia Shi; Xuanlei Zhao; Pengfei Zhou; Kaipeng Zhang; Zhangyang Wang; Kai Wang 036; Yang You 001 (CoRR, 2025)

Drag-and-Drop LLMs: Zero-Shot Prompt-to-Weights. - Zhiyuan Liang; Dongwen Tang; Yuhao Zhou 004; Xuanlei Zhao; Mingjia Shi; Wangbo Zhao; Zekai Li; Peihao Wang; Konstantin Schürholt; Damian Borth; Michael M. Bronstein; Yang You 001; et al. (CoRR, 2025)

HelixPipe: Efficient Distributed Training of Long Sequence Transformers with Attention Parallel Pipeline Parallelism. - Geng Zhang; Shenggan Cheng; Xuanlei Zhao; Ziming Liu; Yang You 001 (CoRR, 2025)