kakaobrain/torchgpipe
A GPipe implementation in PyTorch
862 stars and 301 monthly downloads. No commits in the last 6 months. Available on PyPI.
Stars
862
Forks
98
Language
Python
License
BSD-3-Clause
Category
Last pushed
Jul 25, 2024
Monthly downloads
301
Commits (30d)
0
Dependencies
1
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/kakaobrain/torchgpipe"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related frameworks
deepspeedai/DeepSpeed
DeepSpeed is a deep learning optimization library that makes distributed training and inference...
horovod/horovod
Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
helmholtz-analytics/heat
Distributed tensors and Machine Learning framework with GPU and MPI acceleration in Python
bsc-wdc/dislib
The Distributed Computing library for python implemented using PyCOMPSs programming model for HPC.
google/sedpack
Sedpack - Scalable and efficient data packing