mpi4jax's tracked open-source repos, sorted by stars.
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
