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

9 +0

373 +1

26 +1

GitHub