mattjj/jax-llm-examples

JAX LLM examples

A collection (in progress) of example high-performance large language model implementations, written with JAX.

Current contents include:

• DeepSeek R1