jax-llm-examples

Python ★ 267 updated 8d ago

Minimal yet performant LLM examples in pure JAX

No plain-English explanation yet — one is being written right now. Check back in a minute.