kakaobrain/torchgpipe

A GPipe implementation in PyTorch

/ 100

Established

862 stars and 301 monthly downloads. No commits in the last 6 months. Available on PyPI.

Stale 6m

Maintenance 0 / 25

Adoption 16 / 25

Maturity 25 / 25

Community 20 / 25

Stars

862

Forks

Language

Python

License

BSD-3-Clause

Category

Last pushed

Jul 25, 2024

Monthly downloads

301

Commits (30d)

Dependencies

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