axial-attention
axial-attention copied to clipboard
Is there tensorflow implementations of axial-attention?
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?