kakaobrain/torchgpipe

A GPipe implementation in PyTorch

↗ Open on GitHub

Stars

865

Forks

99

Language

Python

License

BSD-3-Clause

Last push

2 years ago

Python deep-learning pytorch parallelism pipeline-parallelism checkpointing gpipe model-parallelism

No related intel yet

This repo has not appeared in any of the sources the radar tracks. The collector runs on a schedule — check back once it covers this repo.