Visualizing how Axial Attention decomposes 2D attention into Row and Column attention.
Select a mode to start.