main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\legacy\zero\shard_utils\tensor_shard_strategy.py
9 3 -
A
tests\test_legacy\test_amp\test_torch_fp16.py
9 4 -
A
tests\test_analyzer\test_fx\test_mod_dir.py
9 3 -
A
applications\ColossalMoE\utils.py
9 1 -
A
tests\test_legacy\test_amp\test_naive_fp16.py
9 4 -
A
examples\images\diffusion\ldm\modules\midas\midas\dpt_depth.py
9 3 -
A
examples\tutorial\sequence_parallel\model\layers\bert_layer.py
9 4 -
A
examples\language\opt\data.py
9 2 -
A
examples\language\palm\palm_pytorch\autoregressive_wrapper.py
9 2 -
A
tests\test_infer\test_cuda_graph.py
8 1 -
A
tests\test_analyzer\test_fx\test_shape_prop.py
8 4 -
A
colossalai\fx\tracer\bias_addition_patch\patched_bias_addition_function\addbmm.py
8 2 -
A
examples\language\grok-1\grok1_policy.py
8 1 -
A
colossalai\fx\tracer\bias_addition_patch\patched_bias_addition_module\bias_addition_module.py
8 3 -
A
setup.py
8 9 -
A
examples\tutorial\sequence_parallel\data\datasets\test\test_indexed_dataset.py
8 2 1
A
colossalai\inference\core\base_engine.py
8 1 -
A
colossalai\inference\flash_decoding_utils.py
8 1 -
A
colossalai\zero\low_level\zero_hook.py
8 1 -
A
colossalai\inference\modeling\policy\nopadding_baichuan.py
8 4 -