Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
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.
