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/utils
Jiarui Fang c92f84fcdb
[tensor] distributed checkpointing for parameters (#1240)
2 years ago
..
checkpoint [tensor] distributed checkpointing for parameters (#1240) 2 years ago
data_sampler
model [tensor] distributed checkpointing for parameters (#1240) 2 years ago
multi_tensor_apply
profiler
tensor_detector
__init__.py
activation_checkpoint.py
checkpointing.py
common.py [hotfix] fix sharded optim step and clip_grad_norm (#1226) 2 years ago
cuda.py
memory.py
moe.py
timer.py