salesforce/jaxformer

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

by Salesforce Python BSD-3-Clause licence 39 forks Created Aug 2022 View on GitHub

Stars
301
<1% of salesforce's stars
Rank
#196,773
of 852,119 repos, top 24%
In Python
#37,588
of 173,305 repos, top 22%

Among 852,119 repos

Compare with

250–400 stars: 77,415 repos, including jaxformer

Stars along the bottom, and repos in each band on a log scale

Quick stats

Next rank
+1 star to reach #196,208
Next milestone
+55 stars to reach the top 20%
Popular
One of 490,778 repos with 100+ stars
Yearly average
72 stars a year since Aug 2022

Stars counted Recounted daily Next count around