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 8106d7b8c7
[ddp] refactor ColoDDP and ZeroDDP (#1146)
2 years ago
..
amp
builder
cli
communication
context
engine [pipeline]support more flexible pipeline (#1138) 2 years ago
fx
gemini
kernel
logging
nn [ddp] refactor ColoDDP and ZeroDDP (#1146) 2 years ago
pipeline [pipeline]add customized policy (#1139) 2 years ago
registry
tensor [tensor] added __repr__ to spec (#1147) 2 years ago
testing
trainer
utils
zero [ddp] refactor ColoDDP and ZeroDDP (#1146) 2 years ago
__init__.py
constants.py
core.py
global_variables.py
initialize.py