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.
ColossalAI/colossalai
shenggan 5c3843dc98
add colossalai kernel module (#55)
3 years ago
..
amp add interleaved pipeline, fix naive amp and update pipeline model initializer (#80) 3 years ago
builder add interleaved pipeline, fix naive amp and update pipeline model initializer (#80) 3 years ago
communication add interleaved pipeline, fix naive amp and update pipeline model initializer (#80) 3 years ago
context add interleaved pipeline, fix naive amp and update pipeline model initializer (#80) 3 years ago
engine add interleaved pipeline, fix naive amp and update pipeline model initializer (#80) 3 years ago
kernel add colossalai kernel module (#55) 3 years ago
logging update examples and sphnix docs for the new api (#63) 3 years ago
nn overlap computation and communication in 2d operations (#75) 3 years ago
registry Develop/experiments (#59) 3 years ago
trainer added CI for unit testing (#69) 3 years ago
utils add interleaved pipeline, fix naive amp and update pipeline model initializer (#80) 3 years ago
zero fixed zero level 3 dtype bug (#76) 3 years ago
__init__.py Develop/experiments (#59) 3 years ago
constants.py Develop/experiments (#59) 3 years ago
core.py Develop/experiments (#59) 3 years ago
initialize.py add interleaved pipeline, fix naive amp and update pipeline model initializer (#80) 3 years ago