mpi4jax in n2cholas/awesome-jax

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

created at July 21, 2020, 10:57 a.m.

Python

10 +0

379 +3

26 +0

GitHub