multihead-attention
Flexible Python library providing building blocks (layers) for reproducible Transformers research (Tensorflow ✅, Pytorch 🔜, and Jax 🔜)
A Graph Attention Framework for extracting Graph Attention embeddings and implementing Multihead Graph Attention Networks