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 5f2e34e6c9
[booster] implement Gemini plugin (#3352)
2 years ago
..
_C
_analyzer
amp
auto_parallel
autochunk
booster [booster] implement Gemini plugin (#3352) 2 years ago
builder
checkpoint_io
cli
cluster
communication
context
device
engine
fx
gemini
interface
kernel
logging
nn
pipeline
registry
tensor
testing
trainer
utils
zero
__init__.py
constants.py
core.py
global_variables.py
initialize.py