@misc{indiciae2a7d81ab9027, title = {PartIR: Composing SPMD Partitioning Strategies for Machine Learning}, author = {Sami Alabed and Daniel Belov and Bart Chrzaszcz and Juliana Franco and Dominik Grewe and Dougal Maclaurin and James Molloy and Tom Natan and Tamara Norman and Xiaoyue Pan and Adam Paszke and Norman A. Rink and Michael Schaarschmidt and Timur Sitdikov and Agnieszka Swietlik and Dimitrios Vytiniotis and Joel Wee}, year = {2024}, doi = {10.1145/3669940.3707284}, url = {https://arxiv.org/abs/2401.11202}, note = {Source identifier: 2401.11202} }