JaxPP introduces a user-defined MPMD pipeline schedule API on top of JAX/GSPMD and reports throughput gains up to 1.11x over SPMD training on H100 clusters.
Title resolution pending
1 Pith paper cite this work. Polarity classification is still indexing.
1
Pith paper citing it
fields
cs.DC 1years
2024 1verdicts
CONDITIONAL 1representative citing papers
citing papers explorer
-
Scaling Deep Learning Training with MPMD Pipeline Parallelism
JaxPP introduces a user-defined MPMD pipeline schedule API on top of JAX/GSPMD and reports throughput gains up to 1.11x over SPMD training on H100 clusters.