mpi4jax/mpi4jax

JAX 数组的零拷贝 MPI 通信，为 Python 中的高性能计算 (HPC) 应用加速。

↗ 在 GitHub 打开

星数

544

Fork 数

32

语言

Python

许可

MIT

最后推送

2周前

Python jit jax gpu parallel-computing high-performance-computing mpi xla

还没有相关情报

radar 追踪的来源里还没有出现过这个 repo。收集器按计划运行——等它覆盖到这个 repo 再回来看看。