Last released Oct 20, 2026
JAX bindings for NVIDIA AmgX sparse linear solver with automatic differentiation support