Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
Why do you think that https://github.com/dask/dask 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/dask/dask is a good alternative to mpi4jax