GitHub repository: onlyrico/se3-transformer-pytorch
Primary language: Python
Description: Implementation of SE3-Transformers for Equivariant Self-Attention, in Pytorch