Last released Oct 8, 2026
JAXMg provides a jittable C++/CUDA interface between JAX and NVIDIA cuSOLVERMp.