Skip to content

[JAX] Collective GEMM custom op with nvte_cublas_gemm (no comm. overlap) #6417

[JAX] Collective GEMM custom op with nvte_cublas_gemm (no comm. overlap)

[JAX] Collective GEMM custom op with nvte_cublas_gemm (no comm. overlap) #6417

Annotations

1 warning

PyTorch C++

succeeded Dec 5, 2024 in 22s