You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai
Wenhao Chen bb0a668fee
[hotfix] set return_outputs=False in examples and polish code (#5404)
8 months ago
..
_C
_analyzer
accelerator [hotfix] fix typo change MoECheckpintIO to MoECheckpointIO (#5335) 9 months ago
amp
auto_parallel [hotfix] Fix wrong import in meta_registry (#5392) 9 months ago
autochunk
booster [hotfix] set return_outputs=False in examples and polish code (#5404) 8 months ago
checkpoint_io [hotfix] fix typo change MoECheckpintIO to MoECheckpointIO (#5335) 9 months ago
cli [devops] fix extention building (#5427) 9 months ago
cluster
context
device
fx
inference [hotfix] fix typo s/keywrods/keywords etc. (#5429) 9 months ago
interface
kernel
lazy
legacy [hotfix] fix typo s/keywrods/keywords etc. (#5429) 9 months ago
logging
moe [hotfix] fix typo change MoECheckpintIO to MoECheckpointIO (#5335) 9 months ago
nn [hotfix] fix typo s/keywrods/keywords etc. (#5429) 9 months ago
pipeline [hotfix] set return_outputs=False in examples and polish code (#5404) 8 months ago
shardformer [shardformer] fix gathering output when using tensor parallelism (#5431) 8 months ago
tensor [moe] support mixtral (#5309) 10 months ago
testing
utils Merge pull request #5310 from hpcaitech/feature/npu 10 months ago
zero [llama] fix training and inference scripts (#5384) 9 months ago
__init__.py
initialize.py