Name Date Size #Lines LOC

..--

README.mdH A D25-Apr-2025343 84

_IR.pyH A D25-Apr-202547.5 KiB1,244906

__init__.pyH A D25-Apr-2025661 2925

_backward.pyH A D25-Apr-202513.4 KiB371288

_debug.pyH A D25-Apr-2025557 2215

_unflatten.pyH A D25-Apr-2025741 2821

_utils.pyH A D25-Apr-20252.5 KiB10075

microbatch.pyH A D25-Apr-202515.9 KiB470299

schedules.pyH A D25-Apr-202587 KiB2,1631,659

stage.pyH A D25-Apr-202556.6 KiB1,4691,114

README.md

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