[JAX] Collective GEMM custom op with nvte_cublas_gemm (no comm. overlap)
#5521
| Job | Run time |
|---|---|
| 56s | |
| 56s |
nvte_cublas_gemm (no comm. overlap)
#5521
| Job | Run time |
|---|---|
| 56s | |
| 56s |