The base GNN model computes edge embeddings based only on a Gaussian expansion of edge distances.

To better capture 3D geometry, could also embed angles formed by triplets or quadruplets of atoms. A model that incorporates this idea and works quite well is GemNet (Klicpera et al., NeurIPS 2021) see the following figure.