Geometric Deep Learning for Grid Calculation and Optimization