A minimal version of GPT-2 using PyTorch and Gemma 3 using JAX. - View it on GitHub