SOLAR: A Highly Optimized Data Loading Framework for Distributed Training of CNN-based Scientific Surrogates. - Baixi Sun; Xiaodong Yu 001; Chengming Zhang 006; Jiannan Tian; Sian Jin; Kamil Iskra; Tao Zhou; Tekin Bicer; Pete Beckman; Dingwen Tao (CoRR, 2022)

HEAT: A Highly Efficient and Affordable Training System for Collaborative Filtering Based Recommendation on CPUs. - Chengming Zhang 006; Shaden Smith; Baixi Sun; Jiannan Tian; Jonathan Soifer; Xiaodong Yu 001; Shuaiwen Leon Song; Yuxiong He; Dingwen Tao (ICS, 2023)

Benchmarking and In-depth Performance Study of Large Language Models on Habana Gaudi Processors. - Chengming Zhang 006; Baixi Sun; Xiaodong Yu 001; Zhen Xie; Weijian Zheng; Kamil A. Iskra; Pete Beckman; Dingwen Tao (SC Workshops, 2023)

SDP4Bit: Toward 4-bit Communication Quantization in Sharded Data Parallelism for LLM Training. - Jinda Jia; Cong Xie; Hanlin Lu; Daoce Wang; Hao Feng; Chengming Zhang 006; Baixi Sun; Haibin Lin; Zhi Zhang; Xin Liu; Dingwen Tao (NeurIPS, 2024)

FastCLIP: A Suite of Optimization Techniques to Accelerate CLIP Training with Limited Resources. - Xiyuan Wei; Fanjiang Ye; Ori Yonay; Xingyu Chen; Baixi Sun; Dingwen Tao; Tianbao Yang (CoRR, 2024)

GFormer: Accelerating Large Language Models with Optimized Transformers on Gaudi Processors. - Chengming Zhang 006; Xinheng Ding; Baixi Sun; Xiaodong Yu 001; Weijian Zheng; Zhen Xie; Dingwen Tao (CoRR, 2024)

PVII: A pedestrian-vehicle interactive and iterative prediction framework for pedestrian's trajectory. - Qianwen Shen; Shien Huang; Baixi Sun; Xinyu Chen; Dingwen Tao; Huaiyu Wan; Ergude Bao (Appl. Intell., 2024)

COMPSO: Optimizing Gradient Compression for Distributed Training with Second-Order Optimizers. - Baixi Sun; Weijin Liu; J. Gregory Pauloski; Jiannan Tian; Jinda Jia; Daoce Wang; Boyuan Zhang 002; Mingkai Zheng; Sheng Di; Sian Jin; Zhao Zhang 007; Xiaodong Yu 001; et al. (PPoPP, 2025)