JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training - View it on GitHub