main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\amp\naive_amp\grad_scaler\base_grad_scaler.py
9 4 -
A
colossalai\kernel\triton\llama_act_combine_kernel.py
9 2 -
A
tests\test_fx\test_pipeline\test_hf_model\hf_utils.py
9 2 -
A
colossalai\legacy\amp\torch_amp\torch_amp.py
9 4 -
A
examples\tutorial\sequence_parallel\model\layers\bert_layer.py
9 4 -
A
tests\test_legacy\test_amp\test_torch_fp16.py
9 4 -
A
colossalai\legacy\inference\hybridengine\modeling\_utils.py
9 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\bmm_handler.py
9 5 -
A
colossalai\legacy\inference\tensor_parallel\modeling\_utils.py
9 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\binary_elementwise_generator.py
9 3 -
A
colossalai\legacy\nn\layer\parallel_1d\_operation.py
9 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\placeholder_generator.py
9 4 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_shard_option.py
9 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\where_generator.py
9 4 -
A
colossalai\legacy\nn\layer\wrapper\pipeline_wrapper.py
9 3 -
A
tests\test_analyzer\test_fx\test_mod_dir.py
9 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_getitem_handler.py
9 3 -
A
examples\community\roberta\pretraining\utils\logger.py
9 2 -
A
examples\language\palm\palm_pytorch\autoregressive_wrapper.py
9 2 -
A
colossalai\legacy\zero\shard_utils\tensor_shard_strategy.py
9 3 -