Metadata-Version: 2.1
Name: jaxlib
Version: 0.4.17
Summary: XLA library for JAX
Home-page: https://github.com/google/jax
Author: JAX team
Author-email: jax-dev@google.com
License: Apache-2.0
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Requires-Python: >=3.9
Description-Content-Type: text/markdown
License-File: LICENSE.txt
Requires-Dist: scipy >=1.7
Requires-Dist: numpy >=1.22
Requires-Dist: ml-dtypes >=0.2.0
Provides-Extra: cuda11_pip
Requires-Dist: nvidia-cublas-cu11 >=11.11 ; extra == 'cuda11_pip'
Requires-Dist: nvidia-cuda-cupti-cu11 >=11.8 ; extra == 'cuda11_pip'
Requires-Dist: nvidia-cuda-nvcc-cu11 >=11.8 ; extra == 'cuda11_pip'
Requires-Dist: nvidia-cuda-runtime-cu11 >=11.8 ; extra == 'cuda11_pip'
Requires-Dist: nvidia-cudnn-cu11 >=8.8 ; extra == 'cuda11_pip'
Requires-Dist: nvidia-cufft-cu11 >=10.9 ; extra == 'cuda11_pip'
Requires-Dist: nvidia-cusolver-cu11 >=11.4 ; extra == 'cuda11_pip'
Requires-Dist: nvidia-cusparse-cu11 >=11.7 ; extra == 'cuda11_pip'
Provides-Extra: cuda12_pip
Requires-Dist: nvidia-cublas-cu12 ; extra == 'cuda12_pip'
Requires-Dist: nvidia-cuda-cupti-cu12 ; extra == 'cuda12_pip'
Requires-Dist: nvidia-cuda-nvcc-cu12 ; extra == 'cuda12_pip'
Requires-Dist: nvidia-cuda-runtime-cu12 ; extra == 'cuda12_pip'
Requires-Dist: nvidia-cudnn-cu12 >=8.9 ; extra == 'cuda12_pip'
Requires-Dist: nvidia-cufft-cu12 ; extra == 'cuda12_pip'
Requires-Dist: nvidia-cusolver-cu12 ; extra == 'cuda12_pip'
Requires-Dist: nvidia-cusparse-cu12 ; extra == 'cuda12_pip'

# 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/.
