ColossalAI

History

ver217 c415240db6 [nvme] CPUAdam and HybridAdam support NVMe offload (#1360 ) * impl nvme optimizer * update cpu adam * add unit test * update hybrid adam * update docstr * add TODOs * update CI * fix CI * fix CI * fix CI path * fix CI path * fix CI path * fix install tensornvme * fix CI * fix CI path * fix CI env variables * test CI * test CI * fix CI * fix nvme optim __del__ * fix adam __del__ * fix nvme optim * fix CI env variables * fix nvme optim import * test CI * test CI * fix CI		2 years ago
..
amp	[doc] update rst and docstring (#1351 )	2 years ago
builder	[NFC] polish colossalai/builder/builder.py code style (#1265 )	2 years ago
cli	…
communication	[NFC] polish colossalai/communication/collective.py (#1262 )	2 years ago
context	[doc] update rst and docstring (#1351 )	2 years ago
engine	[hotfix] fix PipelineSharedModuleGradientHandler (#1314 )	2 years ago
fx	[fx]add gpt2 passes for pipeline performance test (#1366 )	2 years ago
gemini	[doc] update rst and docstring (#1351 )	2 years ago
kernel	Recover kernal files	2 years ago
logging	…
nn	[nvme] CPUAdam and HybridAdam support NVMe offload (#1360 )	2 years ago
pipeline	[pipeline]add customized policy (#1139 )	2 years ago
registry	Remove duplication registry (#1078 )	3 years ago
tensor	[hotfix] fix no optimizer in save/load (#1363 )	2 years ago
testing	[test] skip tests when not enough GPUs are detected (#1090 )	3 years ago
trainer	fix issue #1080 (#1071 )	3 years ago
utils	[checkpoint] use args, kwargs in save_checkpoint, load_checkpoint (#1368 )	2 years ago
zero	fix zero optim backward_by_grad and save/load (#1353 )	2 years ago
__init__.py	[NFC] polish colossalai/__init__.py code style (#1285 )	2 years ago
constants.py	fix typo in constants (#1027 )	3 years ago
core.py	[Tensor] distributed view supports inter-process hybrid parallel (#1169 )	2 years ago
global_variables.py	…
initialize.py	[hotfix] remove potiential circle import (#1307 )	2 years ago