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
flybird11111 eedaa3e1ef
[shardformer]fix gpt2 double head (#4663)
1 year ago
..
_C
_analyzer
amp
auto_parallel [legacy] move communication and nn to legacy and refactor logger (#4671) 1 year ago
autochunk
booster [pipeline] set optimizer to optional in execute_pipeline (#4630) 1 year ago
checkpoint_io [legacy] move communication and nn to legacy and refactor logger (#4671) 1 year ago
cli [legacy] move communication and nn to legacy and refactor logger (#4671) 1 year ago
cluster [shardformer] support interleaved pipeline (#4448) 1 year ago
context [legacy] move builder and registry to legacy (#4603) 1 year ago
device
fx
interface [zero] hotfix master param sync (#4618) 1 year ago
kernel [legacy] move communication and nn to legacy and refactor logger (#4671) 1 year ago
lazy
legacy [legacy] move communication and nn to legacy and refactor logger (#4671) 1 year ago
logging [legacy] move communication and nn to legacy and refactor logger (#4671) 1 year ago
nn [legacy] move communication and nn to legacy and refactor logger (#4671) 1 year ago
pipeline [legacy] move communication and nn to legacy and refactor logger (#4671) 1 year ago
shardformer [shardformer]fix gpt2 double head (#4663) 1 year ago
tensor [legacy] move communication and nn to legacy and refactor logger (#4671) 1 year ago
testing
utils [legacy] move communication and nn to legacy and refactor logger (#4671) 1 year ago
zero [legacy] move communication and nn to legacy and refactor logger (#4671) 1 year ago
__init__.py
constants.py
core.py
global_variables.py
initialize.py [legacy] move builder and registry to legacy (#4603) 1 year ago