pytorch/torch/distributed/pipelining
Yuanyuan Chen f91899ca6c [2/N] Add strict parameter to Python zip calls (#166257)
This PR adds `strict=True/False` to zip calls in test utils. strict=True is passed when possible.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/166257
Approved by: https://github.com/janeyx99
2025-11-01 00:35:41 +00:00
..
__init__.py [PP] Add DualPipeV schedule (#159591) 2025-08-14 14:58:35 +00:00
_backward.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
_debug.py
_IR.py [1/N] Remove unused loop variables (#166258) 2025-10-30 12:22:25 +00:00
_schedule_visualizer.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
_unflatten.py
_utils.py [PP] Support OVERLAP_F_B computation type (#158978) 2025-08-01 20:22:30 +00:00
microbatch.py [2/N] Add strict parameter to Python zip calls (#166257) 2025-11-01 00:35:41 +00:00
README.md
schedules.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
stage.py [3/N] fix typo in other folders (#166606) 2025-10-30 10:30:40 +00:00

Pipeline Parallelism for PyTorch

torch.distributed.pipelining is a package for implementing pipeline parallelism on your model.

Our documentation is available here.

pipeline_diagram_web