jaxlib
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
To install this package, run one of the following:
Summary
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Last Updated
Oct 3, 2025 at 11:46
License
Apache-2.0
Total Downloads
3.4M
Supported Platforms
GitHub Repository
http://github.com/jax-ml/jax