Graph Neural Networks For The Description Of Solvation Effects In Molecular Dynamics Simulation