TY - RPRT TI - Improving Automatic Parallel Training via Balanced Memory Workload Optimization AU - Yujie Wang AU - Youhe Jiang AU - Xupeng Miao AU - Fangcheng Fu AU - Shenhan Zhu AU - Xiaonan Nie AU - Yaofeng Tu AU - Bin Cui PY - 2024 DO - 10.1109/tkde.2024.3370614 UR - https://arxiv.org/abs/2307.02031 ID - 2307.02031 ER -