AMP: Automatically Finding Model Parallel Strategies with Heterogeneity Awareness

Published in NeurIPS, 2022