main
TransformerEngine
Repository badge with AMinus rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
D
transformer_engine\pytorch\attention.py
307 23 8
A
transformer_engine\pytorch\module\layernorm_linear.py
156 16 3
A
transformer_engine\pytorch\module\linear.py
140 15 3
A
transformer_engine\pytorch\transformer.py
68 14 2
A
transformer_engine\pytorch\module\base.py
134 13 2
B
tests\pytorch\test_fused_attn.py
83 13 7
B
transformer_engine\pytorch\module\layernorm_mlp.py
215 11 3
A
setup.py
115 10 2
F
tests\pytorch\test_onnx_export.py
166 10 119
A
tests\pytorch\test_numerics.py
117 10 -
A
tests\paddle\test_operators.py
57 9 -
A
transformer_engine\pytorch\fp8.py
99 9 -
A
transformer_engine\jax\fused_attn.py
10 9 -
A
transformer_engine\paddle\layer\attention.py
50 8 -
F
transformer_engine\jax\cpp_extensions.py
265 8 24
A
transformer_engine\jax\flax\transformer.py
 8 3
A
tests\paddle\test_layers.py
129 7 2
A
transformer_engine\paddle\csrc\custom_ops.cu
 7 -
A
transformer_engine\common\fused_attn\fused_attn.cpp
129 7 3
A
tests\pytorch\test_sanity.py
99 7 -