GitHub repository: mpi4jax/mpi4jax
Primary language: Python
Description: Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap: