/axial-attention

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

Primary LanguagePythonMIT LicenseMIT

Stargazers

No one’s star this repository yet.