ColossalAI/colossalai
..
amp
auto_parallel
builder
cli
communication
context
device
engine
fx
gemini
kernel
logging
nn
pipeline
registry
tensor
testing
trainer
utils
zero
__init__.py
constants.py
core.py
global_variables.py
initialize.py