ColossalAI/colossalai/nn
HELSON e5ea3fdeef
[gemini] add GeminiMemoryManger (#832)
* refactor StatefulTensor, tensor utilities

* add unitest for GeminiMemoryManager
2022-04-24 13:08:48 +08:00
..
layer [gemini] add GeminiMemoryManger (#832) 2022-04-24 13:08:48 +08:00
loss
lr_scheduler
metric
model
optimizer [zero] improve adaptability for not-shard parameters (#708) 2022-04-11 13:38:51 +08:00
__init__.py
init.py