A reference implementation of GPT-2 in Python, for teaching ML compilers - View it on GitHub