Skip to content

MPI4DL v0.5

Choose a tag to compare

@Quentin-Anthony Quentin-Anthony released this 25 Sep 02:08
· 32 commits to main since this release
e05797a

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.