GitHub repository: pytorch/tau
Primary language: Python
Description: Pipeline Parallelism for PyTorch