ColossalAI

History

HELSON 02b058032d [fx] meta registration compatibility (#3253 ) * [fx] meta registration compatibility * fix error		2 years ago
..
_C	[setup] support pre-build and jit-build of cuda kernels (#2374 )	2 years ago
_analyzer	[FX] refactor experimental tracer and adapt it with hf models (#3157 )	2 years ago
amp	[test] fixed the triton version for testing (#2608 )	2 years ago
auto_parallel	[auto-parallel] add auto-offload feature (#3154 )	2 years ago
autochunk	[autochunk] support vit (#3084 )	2 years ago
booster	[booster] implemented the torch ddd + resnet example (#3232 )	2 years ago
builder	…
checkpoint_io	[booster] implemented the torch ddd + resnet example (#3232 )	2 years ago
cli	[cli] handled version check exceptions (#2848 )	2 years ago
cluster	[booster] implemented the torch ddd + resnet example (#3232 )	2 years ago
communication	…
context	[NFC] polish colossalai/context/process_group_initializer/initializer_2d.py code style (#2726 )	2 years ago
device	[hotfix] add copyright for solver and device mesh (#2803 )	2 years ago
engine	[doc] add deepspeed citation and copyright (#2996 )	2 years ago
fx	[fx] meta registration compatibility (#3253 )	2 years ago
gemini	Add interface for colo tesnor dp size (#3227 )	2 years ago
interface	[booster] implemented the torch ddd + resnet example (#3232 )	2 years ago
kernel	updated flash attention usage	2 years ago
logging	…
nn	[format] applied code formatting on changed files in pull request 2997 (#3008 )	2 years ago
pipeline	[pipeline] Add Simplified Alpa DP Partition (#2507 )	2 years ago
registry	…
tensor	Add interface for colo tesnor dp size (#3227 )	2 years ago
testing	…
trainer	…
utils	[lazyinit] combine lazy tensor with dtensor (#3204 )	2 years ago
zero	[zero] Refactor ZeroContextConfig class using dataclass (#3186 )	2 years ago
__init__.py	…
constants.py	…
core.py	…
global_variables.py	…
initialize.py	Fix False warning in initialize.py (#2456 )	2 years ago