pytorch/torch/distributed/_shard/sharded_optim
..
__init__.py
api.py