yzhwang / mpi4jax Goto Github PK
View Code? Open in Web Editor NEWThis project forked from mpi4jax/mpi4jax
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
Home Page: https://mpi4jax.readthedocs.io/
License: MIT License