Preprints

• How Does Learning Rate Decay Help Modern Neural Networks? arxiv
  • Kaichao You, Mingsheng Long, Jianmin Wang, Michael I. Jordan

Journal Articles

• depyf: Open the Opaque Box of PyTorch Compiler for Machine Learning Researchers JMLR paper Github
  • Kaichao You, Runsheng Bai, Meng Cao, Jianmin Wang, Ion Stoica, Mingsheng Long
  • A PyTorch ecosystem project (see the introduction in PyTorch blog and PyTorch official twitter )
  • Highlighted in the opening keynote of PyTorch conference 2024
• From Big to Small: Adaptive Learning to Partial-Set Domains TPAMI arxiv IEEE Xplore
  • Kaichao You*, Zhangjie Cao, Ziyang Zhang, Jianmin Wang, Mingsheng Long
• Ranking and Tuning Pre-trained Models: A New Paradigm for Exploiting Model Hubs JMLR paper code
  • Kaichao You, Yong Liu, Ziyang Zhang, Jianmin Wang, Michael I. Jordan, Mingsheng Long
• Tianshou: A Highly Modularized Deep Reinforcement Learning Library JMLR arxiv code
  • Jiayi Weng, Huayu Chen, Dong Yan, Kaichao You, Alexis Duburcq, Minghao Zhang, Hang Su, Jun Zhu
  • Github
  • Core developers

Conference Proceedings

• Jenga: Effective Memory Management for Serving LLM with Heterogeneity SOSP 2025
  • Chen Zhang, Kuntai Du, Shu Liu, Woosuk Kwon, Xiangxi Mo, Yufeng Wang, Xiaoxuan Liu, Kaichao You, Zhuohan Li, Mingsheng Long, Jidong Zhai, Joseph Gonzalez, Ion Stoica
• Efficient ConvBN Blocks for Transfer Learning and Beyond ICLR 2024 arxiv code
  • Kaichao You, Guo Qin, Anchang Bao, Meng Cao, Ping Huang, Jiulong Shan, Mingsheng Long
  • This feature has been already integrated into MMCV and PyTorch. It is available starting from PyTorch 2.2.
• LogME: Practical Assessment of Pre-trained Models for Transfer Learning ICML2021 pdf code talk poster
  • Kaichao You, Yong Liu, Jianmin Wang, Mingsheng Long
  • Included in Google Research.
• Co-Tuning for Transfer Learning NeurIPS2020 code pdf poster
  • Kaichao You, Zhi Kou, Mingsheng Long, Jianmin Wang
  • Second Prize of Spectra Review Paper Competition, see the blog
• Stochastic Normalization NeurIPS2020 code pdf
  • Kaichao You*, Zhi Kou, Mingsheng Long, Jianmin Wang
• Towards Accurate Model Selection in Deep Unsupervised Domain Adaptation ICML2019 pdf code poster slides
  • Kaichao You, Ximei Wang, Mingsheng Long, Michael I. Jordan
• Universal Domain Adaptation CVPR2019 pdf code poster
  • Kaichao You, Mingsheng Long, Zhangjie Cao, Jianmin Wang, Michael I. Jordan
• Learning to Transfer Examples for Partial Domain Adaptation CVPR2019 pdf code poster
  • Kaichao You*, Zhangjie Cao, Mingsheng Long, Jianmin Wang, Qiang Yang