JZTXT

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

发布时间 2024-01-07 19:35:31作者: Angry_Panda

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



image


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