mpi4jax
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in
Python341mit
3 months ago
gpuhigh-performance-computingjax
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in
Python341mit
3 months ago
gpuhigh-performance-computingjax