Titan AI LogoTitan AI

axial-attention

387
33
Python

Project Description

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

Project Information

Created on 5/28/2020
Updated on 10/3/2025