main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_fp8\test_fp8_fsdp_comm_hook.py
14 2 -
A
colossalai\legacy\utils\data_sampler\data_parallel_sampler.py
14 2 -
A
colossalai\shardformer\shard\shard_config.py
14 8 -
A
colossalai\kernel\triton\kvcache_copy.py
14 1 -
A
colossalai\legacy\utils\checkpoint\utils.py
14 2 -
A
tests\test_fx\test_profiler\test_profiler_meta_info_prop.py
14 4 -
A
colossalai\auto_parallel\checkpoint\ckpt_solver_chen.py
14 3 -
A
tests\test_zero\test_gemini\test_chunkv2.py
14 6 -
A
colossalai\legacy\context\random\_helper.py
14 2 -
A
applications\Colossal-LLaMA\colossal_llama\dataset\conversation.py
14 2 -
A
examples\images\diffusion\ldm\modules\midas\api.py
14 2 -
A
extensions\cuda_extension.py
14 8 -
A
applications\ColossalChat\coati\distributed\reward\reward_utils.py
14 1 -
A
applications\ColossalChat\coati\distributed\utils.py
14 1 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\softmax_generator.py
14 2 -
A
colossalai\booster\mixed_precision\fp16_torch.py
14 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_embedding_handler.py
14 3 -
A
examples\tutorial\opt\inference\batch.py
14 2 -
A
tests\test_infer\test_models\test_custom_model.py
13 1 -
A
tests\kit\model_zoo\torchrec\torchrec.py
13 5 -