main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
extensions\pybind\softmax\scaled_masked_softmax.cpp
3 1 -
A
colossalai\legacy\tensor\compute_spec.py
3 2 -
A
colossalai\legacy\utils\data_sampler\base_sampler.py
3 2 -
A
colossalai\booster\plugin\dp_plugin_base.py
3 3 -
A
colossalai\legacy\zero\__init__.py
3 2 -
A
examples\tutorial\sequence_parallel\model\layers\mlp.py
3 2 -
A
colossalai\legacy\zero\shard_utils\base_shard_strategy.py
3 2 -
A
colossalai\cli\cli.py
3 4 -
A
tests\test_legacy\test_layers\test_3d\test_3d.py
3 4 -
A
examples\tutorial\sequence_parallel\data\tokenizer\__init__.py
3 2 -
A
colossalai\nn\optimizer\__init__.py
3 6 -
A
examples\language\grok-1\utils.py
3 3 -
A
colossalai\fx\profiler\experimental\profiler.py
3 6 -
A
colossalai\shardformer\modeling\jit.py
3 2 -
A
tests\kit\model_zoo\transformers\mistral.py
3 3 -
A
examples\tutorial\sequence_parallel\loss_func\cross_entropy.py
3 3 -
A
colossalai\fx\profiler\experimental\shard_utils.py
3 2 -
A
colossalai\shardformer\shard\shardformer.py
3 5 -
A
colossalai\tensor\moe_tensor\moe_info.py
2 3 -
A
colossalai\fx\tracer\_symbolic_trace.py
2 3 -