main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\_analyzer\fx\tracer\proxy.py
27 2 -
A
colossalai\_analyzer\fx\tracer\symbolic_trace.py
10 2 -
A
colossalai\_analyzer\fx\tracer\tracer.py
71 3 -
A
colossalai\accelerator\__init__.py
 1 -
A
colossalai\accelerator\api.py
11 2 -
A
colossalai\accelerator\base_accelerator.py
50 1 -
A
colossalai\accelerator\cpu_accelerator.py
41 1 -
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 -