A minimal JAX re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training - View it on GitHub