torchpippy

Pipeline Parallelism for PyTorch

BSD 3 个版本
PiPPy Team
安装
pip install torchpippy
poetry add torchpippy
pipenv install torchpippy
conda install torchpippy
描述

The PiPPy project stands for Pipeline Parallelism for PyTorch. It consists of a compiler and runtime stack for automated parallelism and scaling of PyTorch models. PiPPy partitions the code of the model in a pipelined fashion and enables multiple micro-batches to execute different parts of the model code concurrently. For details, please visit PiPPy's GitHub page.

版本列表
0.2.0 2024-01-26
0.1.1 2023-05-09
0.1.0 2023-04-18