diffdist
Provides differentiable communication primitives such as Send/Recv/Gather etc for pytorch. Autograd can backpropagate through those, making distributed model parallel models easy to implement.
Installation
In a virtualenv (see these instructions if you need to create one):
pip3 install diffdist
Releases
| Version | Released | Bullseye Python 3.9 |
Bookworm Python 3.11 |
Files |
|---|---|---|---|---|
| 0.1 | 2019-07-10 | |||
Issues with this package?
- Search issues for this package
- Package or version missing? Open a new issue
- Something else? Open a new issue