lucidrains / axial-attention

Implementation of Axial attention - attending to multi-dimensional data efficiently
352Updated 3 years ago

Related projects

Alternatives and complementary repositories for axial-attention