Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch - View it on GitHub