main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
.github\workflows\submodule.yml
 1 -
A
.pre-commit-config.yaml
 1 -
A
.readthedocs.yaml
 1 -
A
colossalai\__init__.py
 5 1
A
colossalai\_meta_registrations.py
43 3 3
A
colossalai\amp\__init__.py
5 1 -
A
colossalai\amp\amp_type.py
 1 -
A
colossalai\amp\apex_amp\__init__.py
1 1 -
A
colossalai\amp\apex_amp\apex_amp.py
4 1 -
A
colossalai\amp\naive_amp\__init__.py
7 1 -
A
colossalai\amp\naive_amp\_fp16_optimizer.py
57 1 1
A
colossalai\amp\naive_amp\_utils.py
13 1 -
A
colossalai\amp\naive_amp\grad_scaler\__init__.py
 1 -
A
colossalai\amp\naive_amp\grad_scaler\base_grad_scaler.py
9 1 -
A
colossalai\amp\naive_amp\grad_scaler\constant_grad_scaler.py
2 1 -
A
colossalai\amp\naive_amp\grad_scaler\dynamic_grad_scaler.py
14 1 -
A
colossalai\amp\naive_amp\naive_amp.py
37 1 1
A
colossalai\amp\torch_amp\__init__.py
3 1 -
A
colossalai\amp\torch_amp\_grad_scaler.py
79 2 -
A
colossalai\amp\torch_amp\torch_amp.py
9 1 -