where N i is the neighborhood of all atoms surrounding atom i within a distance cutoff r cut , M t is a learnable function that takes as input the embeddings h t i and h t j of atoms i and j, as well as their edge data, e ij . The embedding of atom i is then updated using the message from Eq. ( ),