GNNs and Beam Dynamics