Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
Why do you think that https://github.com/dfm/extending-jax is a good alternative to mpi4jax
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
Why do you think that https://github.com/dfm/extending-jax is a good alternative to mpi4jax