azzeddineCH/flash-nanoGPT
Jax/Flax re-write of @karpathy 🐐 NanoGPT using some of the common Jax libraries/features (shmap, pallas, jmp, optax, orbax)
PythonMIT
Issues
- 2
Could you add a license to this repo?
#4 opened by jiagaoxiang
Jax/Flax re-write of @karpathy 🐐 NanoGPT using some of the common Jax libraries/features (shmap, pallas, jmp, optax, orbax)
PythonMIT