Suggest an alternative to

mpi4jax

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:

A URL to the alternative repo (e.g. GitHub, GitLab)

Here you can share your experience with the project you are suggesting or its comparison with mpi4jax. Optional.

A valid email to send you a verification link when necessary or log in.