AMP: Automatically Finding Model Parallel Strategies with Heterogeneity Awareness - Li, Dacheng; Wang, Hongyi; Xing, Eric; Zhang, Hao (Advances in Neural Information Processing Systems, 2022)