Last released Jul 17, 2026
JAXMg provides a C++ interface between JAX and cuSolverMg, NVIDIA's multi-GPU linear solver.
Supported by