×
2024
PartIR: Composing SPMD Partitioning Strategies for Machine Learning.
[DOI]
Sami Alabed
,
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