This work was done while Weihua Hu and Muhammed Shuaibi were at Facebook AI Research.
We acknowledge Pytorch [1]} and Pytorch Geometric [2]}.
We thank Ryotatsu Yanagimoto and Zack Ulissi for insightful discussions on physics and chemistry.
