GitHub repository: mlnomadpy/gpt-jax
Primary language: Python
Description: Jax/Flax rewrite of Karpathy's nanoGPT