main
DeepSpeed
Repository badge with D rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
deepspeed\runtime\sparse_tensor.py
11 3 11
A
tests\unit\checkpoint\test_other_optimizer.py
6 3 10
C
deepspeed\ops\adam\fused_adam.py
22 3 28
A
deepspeed\ops\adam\multi_tensor_apply.py
2 3 4
F
deepspeed\runtime\swap_tensor\optimizer_utils.py
98 3 75
A
deepspeed\utils\init_on_device.py
9 3 12
B
deepspeed\utils\mixed_precision_linkage.py
6 3 13
B
deepspeed\utils\logging.py
24 3 23
D
deepspeed\utils\groups.py
53 3 41
B
tests\unit\alexnet_model.py
16 3 22
A
tests\unit\checkpoint\test_latest_checkpoint.py
3 3 6
A
tests\unit\utils\test_init_on_device.py
5 3 7
D
tests\model\Megatron_GPT2\run_func_test.py
48 3 46
B
deepspeed\runtime\zero\tiling.py
39 3 18
B
deepspeed\ops\lamb\fused_lamb.py
29 3 19
A
op_builder\random_ltd.py
8 3 7
A
deepspeed\utils\nvtx.py
1 3 1
C
deepspeed\utils\debug.py
22 3 37
C
deepspeed\autotuning\config.py
9 3 32
A
deepspeed\ops\sparse_attention\__init__.py
 2 2