@misc{indiciae8e65b3257579, title = {Improving Automatic Parallel Training via Balanced Memory Workload Optimization}, author = {Yujie Wang and Youhe Jiang and Xupeng Miao and Fangcheng Fu and Shenhan Zhu and Xiaonan Nie and Yaofeng Tu and Bin Cui}, year = {2024}, doi = {10.1109/tkde.2024.3370614}, url = {https://arxiv.org/abs/2307.02031}, note = {Source identifier: 2307.02031} }