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
10 months ago
..
gemini Merge branch 'main' into sync/npu 10 months ago
init_ctx
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
sharded_param
__init__.py