lucidrains / lie-transformer-pytorch

Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch
MIT License
87 stars 10 forks source link