pytorch

5d6aa99c - Add sharding strategy to fully_shard (#90192)

Add sharding strategy to fully_shard (#90192) Pull Request resolved: https://github.com/pytorch/pytorch/pull/90192 Approved by: https://github.com/awgu, https://github.com/rohan-varma