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/legacy/zero
ver217 148469348a
Merge branch 'main' into sync/npu
11 months ago
..
gemini Merge branch 'main' into sync/npu 11 months ago
init_ctx [misc] update pre-commit and run all files (#4752) 1 year ago
shard_utils [npu] change device to accelerator api (#5239) 11 months ago
sharded_model [npu] change device to accelerator api (#5239) 11 months ago
sharded_optim [misc] update pre-commit and run all files (#4752) 1 year ago
sharded_param [misc] update pre-commit and run all files (#4752) 1 year ago
__init__.py [misc] update pre-commit and run all files (#4752) 1 year ago