main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\kit\model_zoo\diffusers\diffusers.py
3 3 -
A
tests\kit\model_zoo\diffusers\__init__.py
 1 -
A
tests\kit\model_zoo\custom\simple_net.py
5 1 -
A
tests\kit\model_zoo\custom\simple_mlp.py
7 2 -
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
tests\conftest.py
1 1 -
A
setup.py
8 9 -
A
requirements\requirements.txt
 6 -
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 -