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

11 +0

447 +1

30 +0

GitHub