natten (pypi): Neighborhood Attention Extension.
Keywords: machine learning|science|ml|artificial intelligence|ai|cuda|neighborhood-attention|pytorch