mpi4jax

mpi4jax permits to insert MPI operations inside of jit-compiled jax functions, and to perform automatic differentiation through them.

由

bruceshi2021-01-13 收录

推荐

不推荐

更多信息

conda install -c anaconda mpi4jax

标签

根据用户添加的标签生成

暂无标签