Coarse Graining Molecular Dynamics with Graph Neural Networks