Mesh-TensorFlow: Deep Learning for Supercomputers. - Noam Shazeer; Youlong Cheng; Niki Parmar; Dustin Tran; Ashish Vaswani; Penporn Koanantakool; Peter Hawkins; HyoukJoong Lee; Mingsheng Hong; Cliff Young; Ryan Sepassi; Blake A. Hechtman (NeurIPS, 2018)

Image Classification at Supercomputer Scale. - Chris Ying; Sameer Kumar; Dehao Chen; Tao Wang; Youlong Cheng (CoRR, 2018)

GPipe: Efficient Training of Giant Neural Networks using Pipeline Parallelism. - Yanping Huang; Youlong Cheng; Ankur Bapna; Orhan Firat; Dehao Chen; Mia Xu Chen; HyoukJoong Lee; Jiquan Ngiam; Quoc V. Le; Yonghui Wu; Zhifeng Chen (NeurIPS, 2019)

Lingvo: a Modular and Scalable Framework for Sequence-to-Sequence Modeling. - Jonathan Shen; Patrick Nguyen; Yonghui Wu; Zhifeng Chen; Mia Xu Chen; Ye Jia; Anjuli Kannan; Tara N. Sainath; Yuan Cao 007; Chung-Cheng Chiu; Yanzhang He; Jan Chorowski; et al. (CoRR, 2019)

High Resolution Medical Image Analysis with Spatial Partitioning. - Le Hou; Youlong Cheng; Noam Shazeer; Niki Parmar; Yeqing Li; Panagiotis Korfiatis; Travis M. Drucker; Daniel J. Blezek; Xiaodan Song (CoRR, 2019)

Talking-Heads Attention. - Noam Shazeer; Zhenzhong Lan; Youlong Cheng; Nan Ding 002; Le Hou (CoRR, 2020)

Enhanced Exploration in Neural Feature Selection for Deep Click-Through Rate Prediction Models via Ensemble of Gating Layers. - Lin Guan; Xia Xiao; Ming Chen; Youlong Cheng (CoRR, 2021)

Toward Annotator Group Bias in Crowdsourcing. - Haochen Liu; Joseph Thekinen; Sinem Mollaoglu; Da Tang; Ji Yang; Youlong Cheng; Hui Liu 031; Jiliang Tang (ACL (1), 2022)

Monolith: Real Time Recommendation System with Collisionless Embedding Table. - Zhuoran Liu 004; Leqi Zou; Xuan Zou; Caihua Wang; Biao Zhang; Da Tang; Bolin Zhu; Yijie Zhu; Peng Wu; Ke Wang; Youlong Cheng (ORSUM@RecSys, 2022)

Rating Distribution Calibration for Selection Bias Mitigation in Recommendations. - Haochen Liu; Da Tang; Ji Yang; Xiangyu Zhao 001; Hui Liu 031; Jiliang Tang; Youlong Cheng (WWW, 2022)

CowClip: Reducing CTR Prediction Model Training Time from 12 hours to 10 minutes on 1 GPU. - Zangwei Zheng; Pengtai Xu; Xuan Zou; Da Tang; Zhen Li 026; Chenguang Xi; Peng Wu; Leqi Zou; Yijie Zhu; Ming Chen; Xiangzhuo Ding; Fuzhao Xue; et al. (CoRR, 2022)

CowClip: Reducing CTR Prediction Model Training Time from 12 Hours to 10 Minutes on 1 GPU. - Zangwei Zheng; Pengtai Xu; Xuan Zou; Da Tang; Zhen Li 026; Chenguang Xi; Peng Wu; Leqi Zou; Yijie Zhu; Ming Chen; Xiangzhuo Ding; Fuzhao Xue; et al. (AAAI, 2023)