master
apex
Repository badge with AMinus rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
apex\transformer\pipeline_parallel\utils.py
62 2 -
A
apex\contrib\csrc\groupbn\batch_norm_add_relu.h
60 1 -
A
apex\contrib\openfold_triton\fused_adam_swa.py
57 1 3
A
apex\amp\scaler.py
56 1 -
A
apex\amp\frontend.py
56 2 3
A
apex\transformer\functional\fused_softmax.py
55 3 -
A
tests\L0\run_optimizers\test_fused_optimizer.py
55 2 3
A
apex\optimizers\fused_sgd.py
54 1 1
A
tests\L1\common\main_amp.py
54 1 -
A
apex\contrib\sparsity\asp.py
52 3 1
A
csrc\megatron\generic_scaled_masked_softmax.h
52 2 1
A
apex\contrib\csrc\fmha\src\fmha_fprop_kernel_1xN.h
52 2 9
A
tests\L0\run_transformer\test_fused_softmax.py
52 3 -
A
apex\optimizers\fused_adam.py
50 6 1
A
apex\amp\handle.py
49 1 3
A
apex\fp16_utils\fp16util.py
47 2 1
A
apex\transformer\pipeline_parallel\schedules\fwd_bwd_pipelining_without_interleaving.py
47 4 1
A
apex\contrib\sparsity\permutation_search_kernels\channel_swap.py
45 1 -
A
apex\optimizers\fused_mixed_precision_lamb.py
44 2 1
A
apex\transformer\tensor_parallel\mappings.py
42 2 -