/graph-transformer-psiml

Transformer implemented with graph attention network (GAT) layers from PyTorch Geometric

Primary LanguageJupyter Notebook

Stargazers