main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\accelerator\cuda_accelerator.py
43 2 -
A
colossalai\accelerator\npu_accelerator.py
43 2 -
A
colossalai\amp\naive_amp\grad_scaler\__init__.py
 2 -
A
colossalai\amp\naive_amp\grad_scaler\base_grad_scaler.py
9 4 -
A
colossalai\amp\naive_amp\grad_scaler\constant_grad_scaler.py
2 2 -
A
colossalai\amp\naive_amp\grad_scaler\dynamic_grad_scaler.py
16 4 -
A
colossalai\amp\naive_amp\mixed_precision_mixin\__init__.py
 2 -
A
colossalai\amp\naive_amp\mixed_precision_mixin\base.py
5 2 -
A
colossalai\amp\naive_amp\mixed_precision_mixin\bf16.py
5 1 -
A
colossalai\amp\naive_amp\mixed_precision_mixin\fp16.py
11 3 -
A
colossalai\amp\naive_amp\mixed_precision_optimizer.py
54 3 -
A
colossalai\auto_parallel\checkpoint\__init__.py
 2 -
A
colossalai\auto_parallel\checkpoint\build_c_ext.py
 2 -
A
colossalai\auto_parallel\checkpoint\ckpt_solver_base.py
29 4 -
A
colossalai\auto_parallel\checkpoint\ckpt_solver_chen.py
14 3 -
A
colossalai\auto_parallel\checkpoint\ckpt_solver_rotor.c
43 2 -
A
colossalai\auto_parallel\checkpoint\ckpt_solver_rotor.py
85 6 -
A
colossalai\auto_parallel\checkpoint\operation.py
39 3 -
A
colossalai\auto_parallel\meta_profiler\__init__.py
 2 -
A
colossalai\auto_parallel\meta_profiler\constants.py
 4 -