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

github.com/lucidrains/axial-attention

Problem

Audience

Updates

No founder updates yet. Check back soon.