flatironinstitute/jaxmg

JAXMg: A multi-GPU linear solver in JAX

语言：C++ · GitHub 星标：113 · Fork：6

话题：cuda · distributed-computing · jax

在 GitHub 上查看原仓库