Making large AI models cheaper, faster and more accessible
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.
 
 
 
 
 
duanjunwen 5b5fbcff09 [fix] fix hybridparall use_fp8 config 3 weeks ago
..
_C
_analyzer
accelerator [misc] fit torch api upgradation and remove legecy import (#6093) 1 month ago
amp [feat] zerobubble support moehybridplugin; 2 months ago
auto_parallel
autochunk
booster [fix] fix hybridparall use_fp8 config 3 weeks ago
checkpoint_io [checkpointio] fix hybrid plugin model save (#6106) 3 weeks ago
cli
cluster [FP8] rebase main (#5963) 4 months ago
context
device
fx
inference [shardformer] fix linear 1d row and support uneven splits for fused qkv linear (#6084) 1 month ago
interface [feat] update optimizer bwd; ä¸ 2 months ago
kernel [misc] fit torch api upgradation and remove legecy import (#6093) 1 month ago
lazy [fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016) 3 months ago
legacy [fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016) 3 months ago
logging [fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016) 3 months ago
moe [zerobubble] rebase main (#6075) 2 months ago
nn
pipeline Merge branch 'main' into dev/zero_bubble 3 weeks ago
quantization [fp8] add fallback and make compile option configurable (#6092) 1 month ago
shardformer Merge branch 'feature/zerobubble' of github.com:hpcaitech/ColossalAI into dev/zero_bubble 3 weeks ago
tensor [zerobubble] rebase main (#6075) 2 months ago
testing [fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016) 3 months ago
utils [checkpointio] fix hybrid plugin model save (#6106) 3 weeks ago
zero Merge branch 'main' into dev/zero_bubble 3 weeks ago
__init__.py
initialize.py [zerobubble] rebase main (#6075) 2 months ago