This is the official implementation of the paper "Neural Krylov Iteration for Accelerating Linear System Solving" in NeurIPS 2024.