| .. |
|
_composable
|
.
|
2025-09-20 16:09:34 +08:00 |
|
_shard
|
.
|
2025-09-20 16:09:34 +08:00 |
|
_sharded_tensor
|
.
|
2025-09-20 16:09:34 +08:00 |
|
_sharding_spec
|
.
|
2025-09-20 16:09:34 +08:00 |
|
_symmetric_memory
|
.
|
2025-09-20 16:09:34 +08:00 |
|
_tensor
|
.
|
2025-09-20 16:09:34 +08:00 |
|
_tools
|
.
|
2025-09-20 16:09:34 +08:00 |
|
algorithms
|
.
|
2025-09-20 16:09:34 +08:00 |
|
autograd
|
.
|
2025-09-20 16:09:34 +08:00 |
|
benchmarks
|
.
|
2025-09-20 16:09:34 +08:00 |
|
checkpoint
|
.
|
2025-09-20 16:09:34 +08:00 |
|
elastic
|
.
|
2025-09-20 16:09:34 +08:00 |
|
examples
|
.
|
2025-09-20 16:09:34 +08:00 |
|
fsdp
|
.
|
2025-09-20 16:09:34 +08:00 |
|
launcher
|
.
|
2025-09-20 16:09:34 +08:00 |
|
nn
|
.
|
2025-09-20 16:09:34 +08:00 |
|
optim
|
.
|
2025-09-20 16:09:34 +08:00 |
|
pipelining
|
.
|
2025-09-20 16:09:34 +08:00 |
|
rpc
|
.
|
2025-09-20 16:09:34 +08:00 |
|
tensor
|
.
|
2025-09-20 16:09:34 +08:00 |
|
CONTRIBUTING.md
|
.
|
2025-09-20 16:09:34 +08:00 |
|
__init__.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
_checkpointable.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
_composable_state.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
_functional_collectives.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
_functional_collectives_impl.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
_serialization.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
_state_dict_utils.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
argparse_util.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
c10d_logger.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
collective_utils.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
constants.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
device_mesh.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
distributed_c10d.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
launch.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
logging_handlers.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
remote_device.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
rendezvous.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
run.py
|
.
|
2025-09-20 16:09:34 +08:00 |
|
utils.py
|
.
|
2025-09-20 16:09:34 +08:00 |
hailin
c8e8c1e9ff