Owner · jax-ml
jax-ml
jax-ml's tracked open-source repos, sorted by stars.
2 repos
- #1
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
★ 36,241+26Star change over the last 7 days - #2
Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
★ 1,396+23Star change over the last 7 days
