/axial-attention

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

Primary LanguagePythonMIT LicenseMIT

Watchers

No one’s watching this repository yet.