use new pykrylov CG API