Undergrad student, University of Science and Technology of China
1 paper at NeurIPS 2025
Our work introduces a GNN-based method for learning SPAI preconditioners, enabling efficient and robust training across various datasets and outperforming traditional and learning-based approaches in accelerating Conjugate Gradient solvers on GPUs.
