[Distributed] Add a shard helper for creating DTensors#1271

kwen2501 merged 6 commits intomainpytorch/torchchat:mainfrom

dtensor_shardpytorch/torchchat:dtensor_shardCopy head branch name to clipboard