salesforce / jaxformerLinks

Minimal library to train LLMs on TPU in JAX with pjit().
286Updated last year

Alternatives and similar repositories for jaxformer

Users that are interested in jaxformer are comparing it to the libraries listed below

Sorting: