Making large AI models cheaper, faster and more accessible
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.
 
 
 
 
 
Ofey Chan 8706a8c66c
[NFC] polish colossalai/engine/gradient_handler/__init__.py code style (#3329)
2 years ago
..
_C
_analyzer
amp [NFC] polish colossalai/amp/__init__.py code style (#3272) 2 years ago
auto_parallel
autochunk
booster
builder
checkpoint_io
cli [NFC] polish colossalai/cli/benchmark/models.py code style (#3290) 2 years ago
cluster
communication
context [NFC] polish colossalai/context/random/__init__.py code style (#3327) 2 years ago
device
engine [NFC] polish colossalai/engine/gradient_handler/__init__.py code style (#3329) 2 years ago
fx [hotfix] meta_tensor_compatibility_with_torch2 2 years ago
gemini [NFC] polish colossalai/gemini/paramhooks/_param_hookmgr.py code style 2 years ago
interface
kernel
logging
nn [NFC] polish colossalai/nn/_ops/addmm.py code style (#3274) 2 years ago
pipeline
registry
tensor
testing
trainer
utils
zero
__init__.py
constants.py
core.py
global_variables.py [NFC] polish colossalai/global_variables.py code style (#3259) 2 years ago
initialize.py