Jax计算框架的MPI通讯的扩展库 —— mpi4jax

地址：
https://github.com/mpi4jax/mpi4jax

PS. 关于这个library的性能表现并不是很清楚，没有亲自手动安装并测试过。这里只作为链接收藏之用，毕竟这个属于比较小众的python的Jax框架的扩展library。