Automatic Discovery of Composite SPMD Partitioning Strategies in PartIR. - Sami Alabed; Dominik Grewe; Juliana Franco; Bart Chrzaszcz; Tom Natan; Tamara Norman; Norman A. Rink; Dimitrios Vytiniotis; Michael Schaarschmidt (CoRR, 2022)

PartIR: Composing SPMD Partitioning Strategies for Machine Learning. - 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; et al. (CoRR, 2024)

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; et al. (ASPLOS (1), 2025)