GitHub repository: kakaobrain/torchgpipe
Primary language: Python
Description: A GPipe implementation in PyTorch