Häfner, D., & Vicentini, F. (2021). mpi4jax: Zero-copy MPI communication of JAX arrays. Journal of Open Source Software, 6(65). https://doi.org/10.21105/joss.03419