v0.2.1
[checkpoint] checkpoint for ColoTensor Model (#1196)
[hotfix] fix a running error in test_colo_checkpoint.py (#1387)