search

kakaobrain / torchgpipe

A GPipe implementation in PyTorch
https://torchgpipe.readthedocs.io/
BSD 3-Clause "New" or "Revised" License
800 stars 98 forks source link

1 #30

Closed xiyiyia closed 3 years ago