torch-multi-head-attention (pypi): Multi-head attention implemented in PyTorch
Keywords: attention|multi-head|pytorch