GSPMD: General and Scalable Parallelization for ML Computation Graphs – PaperGrep

https://papergrep.dev/paper/gspmd-general-and-scalable-parallelization-for-ml-c17eda