使用 JAX 数组的零拷贝 MPI通信，适用于 Python 中的高性能计算应用：zap
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
