Fix JAX extension build with NVTE_UB_WITH_MPI=1 #2835
+24
−9
Merged
Loading