Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
Primary LanguagePythonMIT LicenseMIT
No one’s watching this repository yet.