ColossalAI/colossalai
Kai Lv 0adca5b688
[launch] Support IPv4 host initialization in launch (#5822)
2024-06-18 19:18:29 +08:00
..
_C Clean up 2024-06-07 09:09:29 +00:00
_analyzer [test] Fix/fix testcase (#5770) 2024-06-03 15:26:01 +08:00
accelerator
amp
auto_parallel
autochunk
booster Support 4d parallel + flash attention (#5789) 2024-06-17 17:40:47 +08:00
checkpoint_io [shardformer] upgrade transformers to 4.39.3 (#5815) 2024-06-14 10:59:33 +08:00
cli
cluster [Feature] Distributed optimizers: Lamb, Galore, CAME and Adafactor (#5694) 2024-05-14 13:52:45 +08:00
context
device [Feature] Distributed optimizers: Lamb, Galore, CAME and Adafactor (#5694) 2024-05-14 13:52:45 +08:00
fx [test] Fix/fix testcase (#5770) 2024-06-03 15:26:01 +08:00
inference [shardformer] upgrade transformers to 4.39.3 (#5815) 2024-06-14 10:59:33 +08:00
interface [Feature] Distributed optimizers: Lamb, Galore, CAME and Adafactor (#5694) 2024-05-14 13:52:45 +08:00
kernel [NFC] Fix code factors on inference triton kernels (#5743) 2024-05-21 22:12:15 +08:00
lazy [lazy] fix lazy cls init (#5720) 2024-05-17 18:18:59 +08:00
legacy [sync] Sync feature/colossal-infer with main 2024-05-20 15:50:53 +00:00
logging
moe
nn [misc] fix dist logger (#5782) 2024-06-05 15:04:22 +08:00
pipeline
quantization
shardformer change 'xxx if xxx else None' to 'xxx or None' 2024-06-18 03:32:42 +00:00
tensor [Feature] Distributed optimizers: Lamb, Galore, CAME and Adafactor (#5694) 2024-05-14 13:52:45 +08:00
testing [misc] Accelerate CI for zero and dist optim (#5758) 2024-06-05 11:25:19 +08:00
utils
zero [gemini] quick fix on possible async operation (#5803) 2024-06-13 10:35:17 +08:00
__init__.py
initialize.py [launch] Support IPv4 host initialization in launch (#5822) 2024-06-18 19:18:29 +08:00