meta-pytorch/autoparallel

An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.

语言：Python · GitHub 星标：98 · Fork：28

在 GitHub 上查看原仓库