Visualization for simple attention and Google's multi-head attention.
Primary LanguageJavaApache License 2.0Apache-2.0