mpi4jax

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:

License

MIT License

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:

Creator

mpi4jax

Related apps