GitHub repository: sdiehl/gpt2-weights
Primary language: Python
Description: A reference implementation of GPT-2 in Python, for teaching ML compilers