/gpt.jax

Generative Pretrained Model (GPT) in JAX. A step by step guide to train LLMs on large datasets from scratch

Primary LanguagePythonMIT LicenseMIT

No issues in this repository yet.