axial-attention (pypi): Axial Attention
Keywords: attention|artificial intelligence|artificial-intelligence|attention-mechanism|deep-learning|pytorch