mpi4jax

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