autodist icon indicating copy to clipboard operation
autodist copied to clipboard

Published 20 hours ago • verified publisher icon petuum

Add Operation Partitioning

Open sangkeun00 opened this issue 4 years ago • 0 comments

System information

  • AutoDist version: 0.7.0
  • Are you willing to contribute it (Yes/No): Yes

Describe the new feature and the current behavior/state Enable operation partitioning as in Mesh Tensorflow to improve parallelization of very large neural networks.

Will this change the current API? How? May add an additional (operation) partitioner to the kernel.

Describe alternatives you've considered

Additional context

sangkeun00 avatar Sep 18 '20 15:09 sangkeun00