Fig. 10: Schematic of the GNN surrogate architecture used for this work. | npj Computational Materials