Is there tensorflow implementations of axial-attention?

lucidrains / axial-attention

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

MIT License

346 stars 30 forks source link

Is there tensorflow implementations of axial-attention? #12

Open henghamao opened 1 year ago

henghamao commented 1 year ago

Thanks for the sharing. This is great job! As tensorflow is another major framework widely used in production environments, is there a tf version for the work?