Skip to content

Releases: OSU-Nowlab/MPI4DL

MPI4DL v0.6

08 Nov 21:55
998d9ad
Compare
Choose a tag to compare

Added support for:

  • Bidirectional and Layer Parallelism (GEMS+LP)
  • Bidirectional and Pipeline Parallelism (GEMS+PP)
  • Spatial, Bidirectional and Layer Parallelism (SP+GEMS+LP)
  • Spatial, Bidirectional and Pipeline Parallelism (SP+GEMS+PP)
  • Support for AmoebaNet and ResNet models
  • Support for different image sizes and custom datasets

MPI4DL v0.5

25 Sep 02:08
e05797a
Compare
Choose a tag to compare

Initial release with distributed training support for:

  • Layer Parallelism (LP)
  • Pipeline Parallelism (PP)
  • Spatial Parallelism (SP)
  • Spatial and Layer Parallelism (SP+LP)
  • Spatial and Pipeline Parallelism (SP+PP)

Currently supports AmoebaNet and ResNet models. A more general model engine is WIP.