/axial-attention

Implementation of Axial attention - attending to multi-dimensional data efficiently

Primary LanguagePythonMIT LicenseMIT

Stargazers