main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\kernel\jit\bias_gelu.py
4 6 -
A
examples\tutorial\auto_parallel\auto_ckpt_batchsize_test.py
4 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_layer_norm_handler.py
4 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_batch_norm_handler.py
4 3 -
A
colossalai\legacy\amp\apex_amp\apex_amp.py
4 2 -
A
examples\tutorial\opt\inference\benchmark\locustfile.py
4 2 -
A
colossalai\legacy\nn\parallel\layers\cache_embedding\base_embedding.py
4 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_default_reshape_handler.py
4 4 -
A
colossalai\legacy\builder\builder.py
4 2 -
A
applications\ColossalChat\examples\inference\web_chatbot\locustfile.py
4 1 -
A
colossalai\auto_parallel\meta_profiler\meta_registry\pooling.py
4 8 -
A
colossalai\legacy\engine\gradient_accumulation\__init__.py
4 2 -
A
colossalai\auto_parallel\meta_profiler\meta_registry\where.py
4 3 -
A
colossalai\auto_parallel\meta_profiler\registry.py
4 2 -
A
colossalai\nn\lr_scheduler\torch.py
4 4 -
A
colossalai\auto_parallel\offload\mem_optimize.py
4 3 -
A
colossalai\legacy\utils\profiler\extention.py
4 2 -
A
examples\inference\benchmark_ops\benchmark_decoding_attn.py
4 1 -
A
tests\test_infer\test_models\test_attention.py
4 1 -
A
tests\test_infer\test_drafter.py
4 1 -