main
DeepSpeed
Repository badge with D rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
deepspeed\utils\init_on_device.py
9 3 12
B
deepspeed\module_inject\containers\gpt2.py
9 5 14
A
deepspeed\runtime\data_pipeline\data_routing\helper.py
9 2 6
A
csrc\aio\py_test\test_ds_aio.py
9 3 12
A
deepspeed\ops\quantizer\quantizer.py
9 2 10
C
deepspeed\autotuning\config.py
9 3 32
A
op_builder\random_ltd.py
8 3 7
A
op_builder\transformer.py
8 3 9
B
tests\unit\runtime\zero\test_zero_context_ancestry.py
8 3 23
B
tests\unit\runtime\sparse_tensor\test_csr.py
8 2 18
A
op_builder\fused_adam.py
8 3 4
A
csrc\includes\ds_transformer_cuda.h
8 2 -
A
op_builder\fused_lamb.py
8 3 9
C
deepspeed\runtime\zero\config.py
8 5 28
A
csrc\sparse_attention\utils.cpp
8 2 -
B
tests\unit\ops\adagrad\test_cpu_adagrad.py
8 3 24
A
tests\unit\ops\transformer\inference\inference_test_utils.py
8 1 9
B
tests\unit\ops\transformer\inference\test_bias_geglu.py
8 4 21
B
tests\unit\inference\test_checkpoint_sharding.py
7 3 22
C
tests\unit\megatron_model.py
7 5 32