Towards Faster Matrix Diagonalization with Graph Isomorphism Networks and the AlphaZero Framework