EasyLM. Easy to use model parallel large language models in JAX/Flax with pjit support on cloud TPU pods.

github.com/TheBloke/EasyLM

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.