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
ver217 708404d5f8
fix pipeline forward return tensors (#176)
3 years ago
..
amp Fixed docstring in colossalai (#171) 3 years ago
builder Fixed docstring in colossalai (#171) 3 years ago
communication Fixed docstring in colossalai (#171) 3 years ago
context Fixed docstring in colossalai (#171) 3 years ago
engine fix pipeline forward return tensors (#176) 3 years ago
kernel Fixed docstring in colossalai (#171) 3 years ago
logging Fixed docstring in colossalai (#171) 3 years ago
nn Fixed docstring in colossalai (#171) 3 years ago
registry
trainer Fixed docstring in colossalai (#171) 3 years ago
utils Fixed docstring in colossalai (#171) 3 years ago
zero Fixed docstring in colossalai (#171) 3 years ago
__init__.py
constants.py Added MoE parallel (#127) 3 years ago
core.py
global_variables.py Fixed docstring in colossalai (#171) 3 years ago
initialize.py Fixed docstring in colossalai (#171) 3 years ago