ColossalAI

History

Baizhou Zhang d99b2c961a [hotfix] fix grad accumulation plus clipping for gemini (#5002 )		1 year ago
..
_C	[setup] support pre-build and jit-build of cuda kernels (#2374 )	2 years ago
_analyzer	[misc] update pre-commit and run all files (#4752 )	1 year ago
amp	[feature] Add clip_grad_norm for hybrid_parallel_plugin (#4837 )	1 year ago
auto_parallel	[misc] update pre-commit and run all files (#4752 )	1 year ago
autochunk	[misc] update pre-commit and run all files (#4752 )	1 year ago
booster	[moe] merge moe into main (#4978 )	1 year ago
checkpoint_io	[checkpointio] hotfix torch 2.0 compatibility (#4824 )	1 year ago
cli	[bug] Fix the version check bug in colossalai run when generating the cmd. (#4713 )	1 year ago
cluster	[hotfix] Fix the bug where process groups were not being properly released. (#4940 )	1 year ago
context	[moe] merge moe into main (#4978 )	1 year ago
device	[misc] update pre-commit and run all files (#4752 )	1 year ago
fx	[misc] update pre-commit and run all files (#4752 )	1 year ago
inference	[Pipeline Inference] Merge pp with tp (#4993 )	1 year ago
interface	[lazy] support from_pretrained (#4801 )	1 year ago
kernel	[moe] merge moe into main (#4978 )	1 year ago
lazy	[doc] add lazy init docs (#4808 )	1 year ago
legacy	[moe] merge moe into main (#4978 )	1 year ago
logging	[misc] update pre-commit and run all files (#4752 )	1 year ago
moe	[moe] merge moe into main (#4978 )	1 year ago
nn	[moe] merge moe into main (#4978 )	1 year ago
pipeline	[Pipeline inference] Combine kvcache with pipeline inference (#4938 )	1 year ago
shardformer	[Pipeline inference] Combine kvcache with pipeline inference (#4938 )	1 year ago
tensor	[moe] merge moe into main (#4978 )	1 year ago
testing	[test] merge old components to test to model zoo (#4945 )	1 year ago
utils	[moe] merge moe into main (#4978 )	1 year ago
zero	[hotfix] fix grad accumulation plus clipping for gemini (#5002 )	1 year ago
__init__.py	[misc] update pre-commit and run all files (#4752 )	1 year ago
initialize.py	[misc] update pre-commit and run all files (#4752 )	1 year ago