Jiarui Fang a1476ea882 [NFC] polish doc style for ColoTensor (#1457)

2022-08-16 09:21:05 +08:00

..

cache_embedding

[NFC] polish doc style for ColoTensor (#1457)

2022-08-16 09:21:05 +08:00

__init__.py

[FAW] export FAW in _ops (#1438)

2022-08-11 13:43:24 +08:00

colo_module.py

[tensor] a shorter shard and replicate spec (#1245)

2022-07-11 15:51:48 +08:00

embedding.py

[tensor] a shorter shard and replicate spec (#1245)

2022-07-11 15:51:48 +08:00

linear.py

[tensor] a shorter shard and replicate spec (#1245)

2022-07-11 15:51:48 +08:00

module_utils.py

[hotfix] fix unit test test_module_spec (#1321)

2022-07-15 14:02:32 +08:00