main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
examples\tutorial\sequence_parallel\model\layers\bert_layer.py
9 4 -
A
examples\tutorial\sequence_parallel\model\layers\embedding.py
9 2 -
A
tests\test_fp8\test_fp8_ddp_comm_hook.py
9 1 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_getitem_handler.py
9 3 -
A
examples\language\palm\palm_pytorch\autoregressive_wrapper.py
9 2 -
A
colossalai\legacy\zero\shard_utils\tensor_shard_strategy.py
9 3 -
A
tests\test_analyzer\test_fx\test_mod_dir.py
9 3 -
A
colossalai\cluster\process_group_manager.py
9 2 -
A
tests\test_legacy\test_amp\test_torch_fp16.py
9 4 -
A
colossalai\fx\tracer\_meta_trace.py
9 5 -
A
colossalai\tensor\padded_tensor\api.py
9 1 -
A
applications\ColossalMoE\utils.py
9 1 1
A
tests\test_infer\test_kernels\cuda\test_kv_cache_memcpy.py
9 2 -
A
examples\tutorial\auto_parallel\auto_ckpt_solver_test.py
9 3 -
A
examples\tutorial\sequence_parallel\model\layers\head.py
9 4 -
A
colossalai\shardformer\policies\sam.py
9 4 2
A
tests\test_optimizer\test_adam_optim.py
10 2 -
A
tests\test_moe\test_moe_ep_tp.py
10 13 -
A
tests\test_fx\test_tracer\test_activation_checkpoint_annotation.py
10 4 -
A
applications\ColossalChat\coati\trainer\callbacks\base.py
10 1 -