People/jax-ml
jax-ml

jax-ml

Organizationbuilder

Pushing back the limits on numerical computing.

github website
HTML
1.3k★
stars on radar
+9
last 7 days
15
public repos
Open source on the radar · 1
jax-ml/
scaling-book
▲ 13/wk

Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs

1.3k185HTML✓ Reviewed
Roles are derived from products, launches and repositories we track — not from self-description.