Name		Name	Last commit message	Last commit date
parent directory ..
README.md		README.md
_IR.py		_IR.py
__init__.py		__init__.py
_backward.py		_backward.py
_debug.py		_debug.py
_schedule_visualizer.py		_schedule_visualizer.py
_unflatten.py		_unflatten.py
_utils.py		_utils.py
microbatch.py		microbatch.py
schedules.py		schedules.py
stage.py		stage.py

README.md

Pipeline Parallelism for PyTorch

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

Our documentation is available here.