xref: /aosp_15_r20/external/pytorch/torch/distributed/pipelining/README.md (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1# Pipeline Parallelism for PyTorch
2
3`torch.distributed.pipelining` is a package for implementing pipeline parallelism on your model.
4
5Our documentation is available [here](https://pytorch.org/docs/main/distributed.pipelining.html).
6
7![pipeline_diagram_web](https://github.com/pytorch/PiPPy/assets/6676466/c93e2fe7-1cd4-49a2-9fd8-231ec9905e0c)
8