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