GNN-Assisted Phase Space Integration with Application to Atomistics