/llama-2-jax

JAX implementation of the Llama 2 model

Primary LanguagePython

Stargazers