GitHub repository: kp-forks/llama-2-jax
Primary language: Python
Description: JAX implementation of the Llama 2 model