Rebuilding GPT2 inference in ~500 lines of commented code