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.

ver217 59f6f573f1 [pipeline] update shardformer policy		1 year ago
..
_C	…
_analyzer	…
amp	[bf16] add bf16 support (#3882)	2 years ago
auto_parallel	[NFC] polish runtime_preparation_pass style (#4266)	1 year ago
autochunk	fix typo colossalai/auto_parallel autochunk fx/passes etc. (#3808)	2 years ago
booster	[zero] support shard optimizer state dict of zero (#4194)	1 year ago
builder	…
checkpoint_io	[checkpointio] Sharded Optimizer Checkpoint for Gemini Plugin (#4302)	1 year ago
cli	fix localhost measurement (#4320)	1 year ago
cluster	[cluster] add process group mesh (#4039)	1 year ago
communication	[NFC] fix: format (#4270)	1 year ago
context	…
device	[format] applied code formatting on changed files in pull request 4152 (#4157)	1 year ago
engine	[nfc]fix ColossalaiOptimizer is not defined (#4122)	1 year ago
fx	[nfc] fix typo colossalai/cli fx kernel (#3847)	2 years ago
interface	[pipeline] refactor 1f1b schedule (#4115)	1 year ago
kernel	[coloattention] fix import error (#4380)	1 year ago
lazy	[lazy] support init on cuda (#4269)	1 year ago
logging	…
nn	[doc] add Series A Funding and NeurIPS news (#4377)	1 year ago
pipeline	[pipeline] build bloom model and policy , revise the base class of policy (#4161)	1 year ago
registry	…
shardformer	[pipeline] update shardformer policy	1 year ago
tensor	[dtensor] fixed readme file name and removed deprecated file (#4162)	1 year ago
testing	Next commit [checkpointio] Unsharded Optimizer Checkpoint for Gemini Plugin (#4141)	1 year ago
trainer	fix typo with colossalai/trainer utils zero (#3908)	2 years ago
utils	[test] remove useless tests (#4359)	1 year ago
zero	[hotfix] fix unsafe async comm in zero (#4404)	1 year ago
__init__.py	…
constants.py	…
core.py	…
global_variables.py	…
initialize.py	[nfc] fix typo colossalai/zero (#3923)	2 years ago