Directory listing for /.local/lib/python3.10/site-packages/torch/distributed/_shard/

• __init__.py
• __pycache__/
• _utils.py
• api.py
• checkpoint/
• common_op_utils.py
• metadata.py
• op_registry_utils.py
• partial_tensor.py
• replicated_tensor.py
• sharded_optim/
• sharded_tensor/
• sharder.py
• sharding_plan/
• sharding_spec/