HYSEN LABS DIRECTORY

jax-ml

Verified repositories, classifications and analysis from the jax-ml organization.

2 curated open-source projects

jax-ml

jax

★ 36,304

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more.

PythonDeveloper Tools3,779 forks

jax-ml

★ 1,430

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

HTML205 forks