main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\legacy\utils\checkpoint\utils.py
14 2 -
A
extensions\cuda_extension.py
14 8 -
A
colossalai\legacy\utils\data_sampler\data_parallel_sampler.py
14 2 -
A
colossalai\booster\mixed_precision\fp16_torch.py
14 2 -
A
examples\tutorial\opt\inference\batch.py
14 2 -
A
tests\test_infer\test_kvcache_manager.py
14 4 -
A
tests\test_fp8\test_fp8_fsdp_comm_hook.py
14 2 -
A
colossalai\shardformer\shard\shard_config.py
14 8 -
A
colossalai\zero\gemini\memory_tracer\memstats_collector.py
15 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_sum_handler.py
15 3 -
A
examples\images\diffusion\ldm\data\lsun.py
15 3 2
A
.github\workflows\scripts\generate_release_draft.py
15 5 -
A
tests\test_auto_parallel\test_tensor_shard\test_metainfo\utils.py
15 9 -
A
examples\images\diffusion\ldm\modules\midas\midas\midas_net_custom.py
15 3 -
A
examples\images\diffusion\scripts\utils.py
15 2 -
A
tests\test_shardformer\test_model\test_shard_mistral.py
15 2 -
A
tests\test_shardformer\test_layer\test_gpt2_qkv_fused_linear_1d.py
15 4 -
A
colossalai\kernel\triton\no_pad_rotary_embedding.py
15 1 1
A
tests\test_lazy\lazy_init_utils.py
15 8 -
A
colossalai\legacy\inference\tensor_parallel\batch_infer_state.py
15 2 -