2025
PartIR: Composing SPMD Partitioning Strategies for Machine Learning.
Proceedings of the 30th ACM International Conference on Architectural Support for Programming Languages and Operating Systems, 2025

2024
PartIR: Composing SPMD Partitioning Strategies for Machine Learning.
CoRR, 2024

2022
Automatic Discovery of Composite SPMD Partitioning Strategies in PartIR.
CoRR, 2022