/nanoGPT-JAX

Repository nanoGPT from @karpathy accelerated with JAX/Flax! The simplest, fastest repository for training/finetuning medium-sized GPTs.

Primary LanguageJupyter NotebookMIT LicenseMIT

Stargazers