Amazon SageMaker Model Parallelism: A General and Flexible Framework for Large Model Training. - Can Karakus; Rahul Huilgol; Fei Wu; Anirudh Subramanian; Cade Daniel; Derya Çavdar; Teng Xu 010; Haohan Chen; Arash Rahnama; Luis Quintela (CoRR, 2021)