Learned Syntax Aware Embeddings Using Equivariant Graph Neural Networks