digger yu 9265f2d4d7 [NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779) ... * fix typo colossalai/autochunk auto_parallel amp * fix typo colossalai/auto_parallel nn utils etc.		2023-05-23 15:28:20 +08:00
..
cache_embedding	[NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779)	2023-05-23 15:28:20 +08:00
__init__.py	[embedding] rename FreqAwareEmbedding -> CachedEmbedding (#1699)	2022-10-13 22:22:27 +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