GitHub repository: Greg-Tarr/tpa_pytorch
Primary language: Python
Description: Simple implementation of Tensor Product Attention from the T6 paper