google-deepmind / simply

git clone https://github.com/google-deepmind/simply.git

Minimal and scalable research codebase in JAX, designed for rapid iteration on frontier research in LLM and other autoregressive models.