pytorch_eigen_solver module