zijie-tian / torchgpipe Goto Github PK
View Code? Open in Web Editor NEWThis project forked from kakaobrain/torchgpipe
A GPipe implementation in PyTorch
Home Page: https://torchgpipe.readthedocs.io/
License: BSD 3-Clause "New" or "Revised" License