ML-Stealer: Stealing Prediction Functionality of Machine Learning Models with Mere Black-Box Access. - Gaoyang Liu; Shijie Wang; Borui Wan; Zekun Wang; Chen Wang 011 (TrustCom, 2021)

Adaptive Message Quantization and Parallelization for Distributed Full-graph GNN Training. - Borui Wan; Juntao Zhao; Chuan Wu 001 (MLSys, 2023)

QSync: Quantization-Minimized Synchronous Distributed Training Across Hybrid Devices. - Juntao Zhao; Borui Wan; Yanghua Peng; Haibin Lin; Yibo Zhu; Chuan Wu 001 (IPDPS, 2024)

POSTER: LLM-PQ: Serving LLM on Heterogeneous Clusters with Phase-Aware Partition and Adaptive Quantization. - Juntao Zhao; Borui Wan; Chuan Wu 001; Yanghua Peng; Haibin Lin (PPoPP, 2024)

LLM-PQ: Serving LLM on Heterogeneous Clusters with Phase-Aware Partition and Adaptive Quantization. - Juntao Zhao; Borui Wan; Yanghua Peng; Haibin Lin; Chuan Wu 001 (CoRR, 2024)

ByteCheckpoint: A Unified Checkpointing System for LLM Development. - Borui Wan; Mingji Han; Yiyao Sheng; Zhichao Lai; Mofan Zhang; Junda Zhang; Yanghua Peng; Haibin Lin; Xin Liu; Chuan Wu 001 (CoRR, 2024)