torchpippy 0.2.0


pip install torchpippy

  Latest version

Released: Jan 26, 2024

Project Links

Meta
Author: PiPPy Team

Classifiers

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.

Wheel compatibility matrix

Platform Python 3
any

Files in release

Extras: None
Dependencies:
torch (>=2.2.0.dev)