Standardize test parametrization and configuration

[voltjia]

pytest output:

============================= test session starts ==============================
platform linux -- Python 3.12.3, pytest-8.4.1, pluggy-1.6.0
benchmark: 5.1.0 (defaults: timer=time.perf_counter disable_gc=False min_rounds=5 min_time=0.000005 max_time=1.0 calibration_precision=10 warmup=False warmup_iterations=100000)
rootdir: /home/huang/ntops
configfile: pyproject.toml
plugins: jaxtyping-0.3.2, benchmark-5.1.0, typeguard-4.4.4, cov-6.2.1, anyio-4.9.0
collected 668 items

tests/test_abs.py ........                                               [  1%]
tests/test_add.py ........                                               [  2%]
tests/test_addmm.py ..                                                   [  2%]
tests/test_bitwise_and.py ................                               [  5%]
tests/test_bitwise_not.py ................                               [  7%]
tests/test_bitwise_or.py ................                                [  9%]
tests/test_bmm.py ..                                                     [ 10%]
tests/test_clamp.py ........                                             [ 11%]
tests/test_cos.py ........                                               [ 12%]
tests/test_div.py .s..s..s..s..s..s..s..s.                               [ 16%]
tests/test_dropout.py ........                                           [ 17%]
tests/test_eq.py ........                                                [ 18%]
tests/test_exp.py ........                                               [ 19%]
tests/test_ge.py ........                                                [ 20%]
tests/test_gelu.py .s.s.s.s.s.s.s.s                                      [ 23%]
tests/test_gt.py ........                                                [ 24%]
tests/test_isinf.py ........                                             [ 25%]
tests/test_isnan.py ........                                             [ 26%]
tests/test_layer_norm.py ............................................... [ 33%]
.................................................                        [ 41%]
tests/test_le.py ........                                                [ 42%]
tests/test_lt.py ........                                                [ 43%]
tests/test_mm.py ..                                                      [ 44%]
tests/test_mul.py ........                                               [ 45%]
tests/test_ne.py ........                                                [ 46%]
tests/test_neg.py ........                                               [ 47%]
tests/test_pow.py ........                                               [ 48%]
tests/test_relu.py ................                                      [ 51%]
tests/test_rms_norm.py ................................................. [ 58%]
...............                                                          [ 60%]
tests/test_rotary_position_embedding.py ................................ [ 65%]
........................................................................ [ 76%]
........................                                                 [ 79%]
tests/test_rsqrt.py ........                                             [ 81%]
tests/test_scaled_dot_product_attention.py ............................. [ 85%]
.................................................                        [ 92%]
tests/test_sigmoid.py ........                                           [ 94%]
tests/test_silu.py ........                                              [ 95%]
tests/test_sin.py ........                                               [ 96%]
tests/test_softmax.py ........                                           [ 97%]
tests/test_sub.py ........                                               [ 98%]
tests/test_tanh.py ........                                              [100%]

================= 652 passed, 16 skipped in 150.05s (0:02:30) ==================