Code for The Annotated Transformer blog post:

http:https://nlp.seas.harvard.edu/annotated-transformer/

Use requirements.txt to install library dependencies with pip:

conda create -n transformer python=3.9
conda activate transformer
pip install -r requirements.txt

The Annotated Transformer is created using jupytext.

Regular notebooks pose problems for source control - cell outputs end up in the repo history and diffs between commits are difficult to examine. Using jupytext, there is a python script (.py file) that is automatically kept in sync with the notebook file by the jupytext plugin.

The python script is committed contains all the cell content and can be used to generate the notebook file. The python script is a regular python source file, markdown sections are included using a standard comment convention, and outputs are not saved. The notebook itself is treated as a build artifact and is not commited to the git repository.

Prior to using this repo, make sure jupytext is installed by following the installation instructions here.

1. After editing AnnotatedTransformer.ipynb, convert it to .py file:

jupytext --output the_annotated_transformer.py AnnotatedTransformer.ipynb

2. To produce the .ipynb notebook file using the markdown source (or .py file):

jupytext --to ipynb the_annotated_transformer.py

To produce the html version of the notebook, run:

# using new version of jupyterlab (>=4.1.5)
pip install pretty-jupyter
jupyter nbconvert --to html the_annotated_transformer.ipynb --no-prompt --template pj --embed-images

To keep the code formatting clean, the annotated transformer git repo has a git action to check that the code conforms to PEP8 coding standards.

To make this easier, there are two Makefile build targets to run automatic code formatting with black and flake8.

Be sure to install black and flake8.

You can then run:

make black

(or alternatively manually call black black --line-length 79 the_annotated_transformer.py) to format code automatically using black and:

make flake

(or manually call flake8 `flake8 --show-source the_annotated_transformer.py) to check for PEP8 violations.

It's recommended to run these two commands and fix any flake8 errors that arise, when submitting a PR, otherwise the github actions CI will report an error.