/nanogpt-jax

An implementation of nanogpt in jax from scratch based on Andrej Karpathy's tutorial series on Neural Networks.

Primary LanguagePythonMIT LicenseMIT

Stargazers