mpi4jax
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python 
423 stars
11 watching
29 forks
Language: Python
last commit: 5 days ago
Linked from 1 awesome list
gpuhigh-performance-computingjaxjitmpiparallel-computingxla
n2cholas/awesome-jax