Add edge features to Salt

Adding compatibility for edge features in the transformer architectures within Salt based on a slightly modified version of https://arxiv.org/pdf/2108.03348v1.pdf