Minimal and scalable research codebase in JAX, designed for rapid iteration on frontier research in autoregressive models. - View it on GitHub
Star
21
Rank
877041