Skip to content

JAXMg¤

Title Title

JAXMg provides a C++ interface between JAX and cuSolverMg, NVIDIA’s multi-GPU linear solver. We provide a jittable API for the following routines.

  • cusolverMgPotrs: Solves the system of linear equations: \(Ax=b\) where \(A\) is an \(N\times N\) symmetric (Hermitian) positive-definite matrix via a Cholesky decomposition
  • cusolverMgPotrs: Computes the inverse of an \(N\times N\) symmetric (Hermitian) positive-definite matrix via a Cholesky decomposition.
  • cusolverMgPotrs: Computes eigenvalues and eigenvectors of an \(N\times N\) symmetric (Hermitian) matrix.

For more details, see the API and the accompanying paper.