master
apex
Repository badge with AMinus rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
apex\contrib\multihead_attn\fast_encdec_multihead_attn_func.py
3 1 -
A
csrc\multi_tensor_lamb_stage_2.cu
 1 -
A
apex\amp\_amp_state.py
13 2 -
A
csrc\multi_tensor_scale_kernel.cu
 2 -
A
apex\contrib\multihead_attn\fast_self_multihead_attn_func.py
9 1 -
A
csrc\static_switch.h
0 1 -
A
apex\contrib\csrc\group_norm\group_norm_nhwc_one_pass_48.cu
 1 -
A
csrc\type_shim.h
20 1 -
A
apex\contrib\multihead_attn\mask_softmax_dropout_func.py
7 1 -
A
csrc\welford.cu
 1 2
A
apex\amp\scaler.py
56 1 -
A
tests\L0\run_amp\test_add_param_group.py
35 2 1
A
setup.py
10 13 -
A
examples\simple\distributed\run.sh
 1 -
A
apex\_autocast_utils.py
7 1 -
A
apex\amp\frontend.py
56 2 3
A
apex\contrib\conv_bias_relu\__init__.py
 2 -
A
apex\contrib\csrc\focal_loss\focal_loss_cuda_kernel.cu
 1 -
A
apex\contrib\csrc\multihead_attn\self_multihead_attn_bias_additive_mask_cuda.cu
 2 -
A
apex\multi_tensor_apply\__init__.py
 1 -