---------

Co-authored-by: Hongyi Jin <jinhongyi02@gmail.com>
Co-authored-by: Ruihang Lai <ruihangl@cs.cmu.edu>
Co-authored-by: Tianqi Chen <tqchen@cmu.edu>
Co-authored-by: Junru Shao <junrushao@apache.org>
Co-authored-by: Zihao Ye <zhye@cs.washington.edu>