A Graph Neural Network Framework for Grid-Based Simulation