After a bit more digging, it seems Bellard also has a version using GPT-2: https://bellard.org/libnc/gpt2tc.html Looks much cleaner and faster than my little winter break project, cool to see this implemented properly!