Learning Algebraic Multigrid Using Graph Neural Networks