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:
jaxlib is the support library for JAX. While JAX itself is a pure Python package, jaxlib contains the binary (C/C++) parts of the library, including Python bindings, the XLA compiler, the PJRT runtime, and a handful of handwritten kernels.
Summary
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Last Updated
Mar 31, 2025 at 22:02
License
Apache-2.0
Total Downloads
0
Version Downloads
0
Supported Platforms
GitHub Repository
https://github.com/google/jaxDocumentation
https://jax.readthedocs.io/