JacksonFurrier/mpi4jax
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
PythonMIT
No issues in this repository yet.
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
PythonMIT
No issues in this repository yet.