GitHub / mpi4jax 2 Repositories
mpi4jax/mpi4jax
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
Language: Python - Size: 5.06 MB - Last synced at: 13 days ago - Pushed at: 3 months ago - Stars: 476 - Forks: 31
mpi4jax/mpi4jax-joss
JOSS paper for mpi4jax
Language: TeX - Size: 244 KB - Last synced at: about 1 year ago - Pushed at: almost 4 years ago - Stars: 0 - Forks: 0