TY - RPRT TI - GSPMD: General and Scalable Parallelization for ML Computation Graphs AU - Yuanzhong Xu AU - HyoukJoong Lee AU - Dehao Chen AU - Blake Hechtman AU - Yanping Huang AU - Rahul Joshi AU - Maxim Krikun AU - Dmitry Lepikhin AU - Andy Ly AU - Marcello Maggioni AU - Ruoming Pang AU - Noam Shazeer AU - Shibo Wang AU - Tao Wang AU - Yonghui Wu AU - Zhifeng Chen PY - 2021 UR - https://arxiv.org/abs/2105.04663 ID - 2105.04663 ER -