main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\gemini\ophooks\utils.py
20 1 -
A
colossalai\context\process_group_initializer\initializer_tensor.py
6 1 -
A
colossalai\gemini\paramhooks\_param_hookmgr.py
10 1 -
A
colossalai\amp\apex_amp\__init__.py
1 1 -
A
colossalai\gemini\stateful_tensor.py
33 1 -
A
colossalai\context\random\__init__.py
 1 -
A
colossalai\gemini\tensor_placement_policy.py
24 1 -
A
colossalai\auto_parallel\solver\op_handler\layer_norm_handler_v2.py
3 1 -
A
colossalai\global_variables.py
5 1 -
A
colossalai\auto_parallel\solver\op_handler\node_handler.py
14 6 -
A
colossalai\kernel\__init__.py
 1 -
A
colossalai\kernel\cuda_native\csrc\cpu_adam.h
10 2 -
A
colossalai\kernel\cuda_native\csrc\layer_norm_cuda_kernel.cu
 1 -
A
colossalai\kernel\cuda_native\csrc\layer_norm_cuda.cpp
13 1 -
A
colossalai\kernel\cuda_native\csrc\kernels\include\cross_entropy_layer.h
2 1 -
A
colossalai\kernel\cuda_native\csrc\kernels\general_kernels.cu
 1 -
A
.github\workflows\assign_reviewer.yml
 1 -
A
.github\workflows\release_docker.yml
 2 -
A
colossalai\auto_parallel\solver\op_handler\strategy_generator.py
3 2 -
A
colossalai\engine\gradient_handler\_base_gradient_handler.py
2 1 -