jax-mljax★ 36,304Python+NumPy プログラムのコンポーザブル変換: 微分、ベクトル化、JIT から GPU/TPU など。Python開発ツールフォーク 3,779Python開発ツールライブラリクロスプラットフォーム
jax-mlscaling-book★ 1,430Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUsHTMLフォーク 205