where W 3 is a learned weight matrix. The learned atomic (h h h α,i ) and bond (h h h β,ij ) embedding from messaging passing can be used alone or optionally combined with additional QM or non-QM features in a way similar to the initial feature vectors shown in Equation with an additional linear layer, and therefore: