ayaka14732 / llama-jax

JAX implementation of LLaMA, aiming to train LLaMA on Google Cloud TPU
14Updated last year

Related projects

Alternatives and complementary repositories for llama-jax