The model architecture starts from a position-included graph, and then goes through a featurization process, followed by main blocks, and the readout module with energy, force and stress outputs. The ...