/tinyGPT-jax

A tiny and fast GPT implement with JAX

Primary LanguagePython

Watchers