bindog / pytorch-model-parallel

A memory balanced and communication efficient FullyConnected layer with CrossEntropyLoss model parallel implementation in PyTorch
85Updated 4 years ago

Alternatives and similar repositories for pytorch-model-parallel:

Users that are interested in pytorch-model-parallel are comparing it to the libraries listed below