/mnt/ServicesNVME/odysseus/data/local/lib/python3.14/site-packages/torch/distributed/_shard/

6 directories 7 files
Name Size Modified
Go up
__pycache__/
checkpoint/
sharded_optim/
sharded_tensor/
sharding_plan/
sharding_spec/
__init__.py 87 B
_utils.py 1.0 KiB
api.py 12 KiB
common_op_utils.py 2.1 KiB
metadata.py 2.1 KiB
op_registry_utils.py 1.0 KiB
sharder.py 901 B