Directory listing for /docker/overlay2/umogtwqr3ekkctiwz0q72yia0/diff/venv/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
• sharded_optim/
• sharded_tensor/
• sharder.py
• sharding_plan/
• sharding_spec/