PartIR: Composing SPMD Partitioning Strategies for Machine Learning
Sami Alabed, Daniel Belov, Bart Chrzaszcz,Juliana Franco,Dominik Grewe,Dougal Maclaurin,James Molloy, Tom Natan,Tamara Norman, Xiaoyue Pan,Adam Paszke,Norman A. Rink,Michael Schaarschmidt, Timur Sitdikov, Agnieszka Swietlik,Dimitrios Vytiniotis, Joel Wee CoRR(2024)
AI 理解论文
溯源树
样例