where hu and hv are the node embeddings and e is the edge embedding from the previous layer, ne is the length of the edge embedding. OrbNet has different matrices W in different layers. This is a flexible framework for learning orbital interactions and could be naturally transferred to downstream tasks. We implement this model and further, it will be noted as the 'OFMNet' model.