pytorch/torch/distributed/pipelining
2025-02-28 07:35:56 +00:00
..
__init__.py [pipelining] Add ZBV schedule (#142084) 2024-12-11 02:00:57 +00:00
_backward.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
_debug.py remove allow-untyped-defs from torch/distributed/pipelining/_debug.py (#143871) 2024-12-27 01:20:26 +00:00
_IR.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
_unflatten.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00
_utils.py PEP585: More UP006 fixes (#146392) 2025-02-20 06:18:13 +00:00
microbatch.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
README.md
schedules.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
stage.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +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