axial-attention

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

MIT License

Downloads
2.9K
Stars
348
Committers
1