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