JAX LLM examples A collection (in progress) of example high-performance large language model implementations, written with JAX. Current contents include: DeepSeek R1