master
apex
Repository badge with AMinus rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
apex\transformer\tensor_parallel\cross_entropy.py
5 1 -
A
apex\transformer\tensor_parallel\utils.py
5 1 -
A
apex\contrib\openfold_triton\_layer_norm_backward_kernels.py
5 1 -
A
tests\L0\run_transformer\test_cross_entropy.py
5 1 -
A
apex\multi_tensor_apply\multi_tensor_apply.py
5 1 -
A
apex\contrib\test\multihead_attn\test_self_multihead_attn.py
5 2 -
A
apex\contrib\test\multihead_attn\test_encdec_multihead_attn.py
5 2 -
A
apex\contrib\csrc\layer_norm\ln.h
5 3 3
A
apex\contrib\test\focal_loss\test_focal_loss.py
5 1 -
A
docs\source\conf.py
4 1 2
A
apex\transformer\utils.py
4 2 -
A
apex\transformer\pipeline_parallel\schedules\__init__.py
4 1 -
A
apex\contrib\csrc\fmha\src\fmha\mask.h
4 1 2
A
apex\transformer\functional\fused_rope.py
4 2 -
A
tests\L0\run_transformer\test_transformer_utils.py
4 1 -
A
apex\contrib\test\openfold_triton\test_openfold_mha.py
4 1 -
A
csrc\megatron\scaled_softmax.cpp
3 2 -
A
csrc\megatron\generic_scaled_masked_softmax.cpp
3 2 1
A
csrc\megatron\fused_rotary_positional_embedding.cpp
3 2 -
A
csrc\megatron\scaled_upper_triang_masked_softmax.cpp
3 2 -