GitHub repository: prameth/bertviz
Primary language: Jupyter Notebook
Description: Tool for visualizing attention in the Transformer model (BERT, GPT-2, Albert, XLNet, RoBERTa, CTRL, etc.)