__init__.py
__pycache__/
_api.py
_argmin_argmax.py
_collective_utils.py
_dispatch.py
_dtensor_spec.py
_op_schema.py
_ops/
_random.py
_redistribute.py
_sharding_prop.py
_shards_wrapper.py
_tp_conv.py
_utils.py
debug/
device_mesh.py
experimental/
parallel/
placement_types.py