/minGPT-jax

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

Primary LanguagePythonMIT LicenseMIT

No issues in this repository yet.