Jax/Flax re-write of @karpathy π NanoGPT using some of the common Jax libraries/features (shmap, pallas, jmp, optax, orbax)
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool