jaxformer

Python ★ 299 updated 18d ago

Minimal library to train LLMs on TPU in JAX with pjit().

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