Fig. 5: Example architecture for deep learning models. | npj Computational Materials