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

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Core	26m 31s
JAX	47m 3s
PaddlePaddle	23m 29s
PyTorch	59m 17s
	2h 36m 20s