ColossalAI/colossalai
Xuanlei Zhao 68fcaa2225
remove duplicate import (#5100)
2023-11-23 15:15:01 +08:00
..
_C
_analyzer
amp
auto_parallel
autochunk
booster [npu] add npu support for hybrid plugin and llama (#5090) 2023-11-22 19:23:21 +08:00
checkpoint_io
cli
cluster
context
device [npu] add npu support for hybrid plugin and llama (#5090) 2023-11-22 19:23:21 +08:00
fx
inference [Hotfix] Fix model policy matching strategy in ShardFormer (#5064) 2023-11-22 11:19:39 +08:00
interface
kernel
lazy
legacy [npu] add npu support for hybrid plugin and llama (#5090) 2023-11-22 19:23:21 +08:00
logging
moe
nn
pipeline
shardformer remove duplicate import (#5100) 2023-11-23 15:15:01 +08:00
tensor
testing [npu] add npu support for hybrid plugin and llama (#5090) 2023-11-22 19:23:21 +08:00
utils [npu] add npu support for hybrid plugin and llama (#5090) 2023-11-22 19:23:21 +08:00
zero
__init__.py
initialize.py