Rare find

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

github.com/salesforce/jaxformer
by @salesforce

Problem

Audience

Keep exploring

Updates

No founder updates yet. Check back soon.