[WIP] Use DTensor-based tensor parallel#180

kwen2501 wants to merge 1 commit intogh/kwen2501/1/basemeta-pytorch/gpt-fast:gh/kwen2501/1/basefrom

gh/kwen2501/1/headmeta-pytorch/gpt-fast:gh/kwen2501/1/headCopy head branch name to clipboard