Project: jaxlib

XLA library for JAX

Project Details

Latest version
0.4.23
Home Page
https://github.com/google/jax
PyPI Page
https://pypi.org/project/jaxlib/

Project Popularity

PageRank
0.01167222001078482
Number of downloads
2280359

jaxlib: support library for JAX

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. For more information, including installation and build instructions, refer to main JAX README: https://github.com/google/jax/.