RÉPERTOIRE HYSEN LABS

jax-ml

Dépôts vérifiés, classifications et analyses de l’organisation jax-ml.

2 projets open source sélectionnés

jax-ml

jax

★ 36 304

Transformations composables de programmes Python+NumPy : différenciation, vectorisation, JIT vers GPU/TPU, et plus encore.

PythonOutils de développement3 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