pytorch/torch/distributed/_tensor
..
__init__.py
api.py
placement_types.py