[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

transformer_engine/jax/gemm.py

@@ -151,7 +151,6 @@ def _gemm_fwd_rule(
 
     fuse_bias = bias is not None
 
-
     # AG+GEMM: ([B], M/P, K) --(AG)--> ([B], M, K) x (K, N/P) --------> ([B], M, N/P)
     # GEMM+AR:                       ([B], M, K/P) x (K/P, N) --(AR)--> ([B], M, N)
     out, pre_gelu_out, extra_out = gemm_impl(
@@ -204,7 +203,6 @@ def _gemm_bwd_rule(
         mirror_dim, (x_inner_dim, kernel_inner_dim), (x.ndim, kernel.ndim)
     )
 
-
     dgrad_overlap_config = None
     if comm_overlap_config is not None:
         dgrad_overlap_name = comm_overlap_config["name"].rstrip("_fprop") + "_dgrad"