local-attention-flax (pypi): Local Attention - Flax Module in Jax
Keywords: artificial intelligence|deep learning|attention mechanism|jax|artificial-intelligence|attention-mechanism|deep-learning