Axial Attention in Multidimensional Transformers

Benchmark Model Rank Results
image-generation-on-imagenet-64x64Axial Transformer (6 layers)#50Bits per dim: 4.032