Yuliang Liu, Shenggui Li, Jiarui Fang, Yanjun Shao, Boyuan Yao, Yang You: Colossal-Auto: Unified Automation of Parallelization and Activation Checkpoint for Large-scale Models. CoRR abs/2302.02599 (2023)