main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\kit\model_zoo\custom\simple_net.py
5 1 -
A
tests\kit\model_zoo\custom\simple_mlp.py
7 1 -
A
tests\kit\model_zoo\custom\repeated_computed_layers.py
6 1 -
A
tests\kit\model_zoo\custom\nested_model.py
7 1 -
A
tests\kit\model_zoo\custom\hanging_param_model.py
5 1 -
A
tests\kit\model_zoo\custom\base.py
6 1 -
A
tests\kit\model_zoo\custom\__init__.py
 2 -
A
tests\kit\model_zoo\__init__.py
 6 -
A
setup.py
8 9 -
A
requirements\requirements.txt
 4 -
A
extensions\utils.py
33 2 -
A
extensions\triton_extension.py
4 2 -
A
extensions\pybind\softmax\scaled_upper_triangle_masked_softmax_cuda.py
5 1 -
A
extensions\pybind\softmax\scaled_upper_triang_masked_softmax.cpp
3 1 -
A
extensions\pybind\softmax\scaled_masked_softmax_cuda.py
5 1 -
A
extensions\pybind\softmax\scaled_masked_softmax.cpp
3 1 -
A
extensions\pybind\softmax\__init__.py
 1 -
A
extensions\pybind\optimizer\optimizer.cpp
1 1 -
A
extensions\pybind\optimizer\fused_optimizer_cuda.py
5 1 -
A
extensions\pybind\optimizer\__init__.py
 1 -