Tool for visualizing attention in the Transformer model (BERT, GPT-2, Albert, XLNet, RoBERTa, CTRL, etc.)
Primary LanguageJupyter NotebookApache License 2.0Apache-2.0