Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Function transformations over NumPy-style arrays instead of a graph-and-layers API.
- 言語
- Python
- ライセンス
- Apache-2.0
- スター
- 36k
- 最新リリース
- jax-v0.11.2
- 最終プッシュ
- 2026-10-10