main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
extensions\pybind\moe\moe_cuda.py
5 1 -
A
colossalai\amp\naive_amp\mixed_precision_mixin\base.py
5 2 -
A
colossalai\amp\naive_amp\mixed_precision_mixin\bf16.py
5 1 -
A
colossalai\legacy\amp\__init__.py
5 2 -
A
examples\language\gpt\gemini\commons\utils.py
5 3 -
A
tests\kit\model_zoo\transformers\bloom.py
5 6 -
A
extensions\pybind\inference\inference_ops_cuda.py
5 1 -
A
examples\language\gpt\experiments\auto_offload\model_zoo.py
5 2 -
A
tests\kit\model_zoo\custom\simple_net.py
5 1 -
A
colossalai\nn\lr_scheduler\multistep.py
5 5 -
A
colossalai\legacy\global_variables.py
5 2 -
A
colossalai\legacy\inference\dynamic_batching\get_tokenizer.py
5 1 -
A
colossalai\legacy\inference\dynamic_batching\ray_init_config.py
5 1 -
A
colossalai\legacy\inference\quant\gptq\cai_gptq\gptq_op.py
5 1 -
A
examples\images\diffusion\ldm\modules\midas\midas\midas_net.py
5 3 -
A
tests\kit\model_zoo\custom\hanging_param_model.py
5 1 -
A
colossalai\auto_parallel\tensor_shard\node_handler\output_handler.py
5 4 -
A
colossalai\shardformer\shard\utils.py
5 1 -
A
colossalai\legacy\nn\layer\colossalai_layer\dropout.py
5 3 -
A
colossalai\legacy\nn\layer\colossalai_layer\linear.py
5 2 -