mpi4jax in n2cholas/awesome-jax

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

updated at May 26, 2024, 6:38 p.m.

Python

10 +0

379 +3

26 +0

GitHub