xla
e60428d1 - [SPMD] suppor DTensor API integration (#5776)

Commit
1 year ago
[SPMD] suppor DTensor API integration (#5776) * [SPMD] move SPMD package to torch_xla/experimental/spmd, introduce shadow xla DTensor API. * support backward compatibility of the old imports * Move spmd out of experimental * Update spmd.md for distributed/spmd
Author
Parents
Loading