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
Hongxin Liu d4a436051d
[checkpointio] support async model save (#6131)
5 days ago
..
_C
_analyzer
accelerator [misc] fit torch api upgradation and remove legecy import (#6093) 1 month ago
amp [plugin] support get_grad_norm (#6115) 3 weeks ago
auto_parallel
autochunk
booster [checkpointio] support async model save (#6131) 5 days ago
checkpoint_io [checkpointio] support async model save (#6131) 5 days ago
cli [cli] support run as module option (#6135) 1 week ago
cluster
context
device
fx
inference [shardformer] fix linear 1d row and support uneven splits for fused qkv linear (#6084) 2 months ago
interface [plugin] support get_grad_norm (#6115) 3 weeks ago
kernel [misc] fit torch api upgradation and remove legecy import (#6093) 1 month ago
lazy
legacy
logging
moe
nn
pipeline [misc] fit torch api upgradation and remove legecy import (#6093) 1 month ago
quantization [fp8] add fallback and make compile option configurable (#6092) 1 month ago
shardformer [hotfix] fix flash attn window_size err (#6132) 1 week ago
tensor
testing
utils [checkpointio] support async model save (#6131) 5 days ago
zero [fix] multi-node backward slowdown (#6134) 1 week ago
__init__.py
initialize.py [fp8] hotfix backward hook (#6053) 2 months ago