Metadata-Version: 2.1
Name: jax-cuda12-plugin
Version: 0.7.0
Summary: JAX Plugin for NVIDIA GPUs
Home-page: https://github.com/jax-ml/jax
Author: JAX team
Author-email: jax-dev@google.com
License: Apache-2.0
Classifier: Development Status :: 5 - Production/Stable
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Programming Language :: Python :: Free Threading :: 3 - Stable
Requires-Python: >=3.11
Description-Content-Type: text/markdown
Requires-Dist: jax-cuda12-pjrt ==0.7.0
Provides-Extra: with-cuda
Requires-Dist: nvidia-cublas-cu12 >=12.1.3.1 ; extra == 'with-cuda'
Requires-Dist: nvidia-cuda-cupti-cu12 >=12.1.105 ; extra == 'with-cuda'
Requires-Dist: nvidia-cuda-nvcc-cu12 >=12.6.85 ; extra == 'with-cuda'
Requires-Dist: nvidia-cuda-runtime-cu12 >=12.1.105 ; extra == 'with-cuda'
Requires-Dist: nvidia-cudnn-cu12 <10.0,>=9.8 ; extra == 'with-cuda'
Requires-Dist: nvidia-cufft-cu12 >=11.0.2.54 ; extra == 'with-cuda'
Requires-Dist: nvidia-cusolver-cu12 >=11.4.5.107 ; extra == 'with-cuda'
Requires-Dist: nvidia-cusparse-cu12 >=12.1.0.106 ; extra == 'with-cuda'
Requires-Dist: nvidia-nccl-cu12 >=2.18.1 ; extra == 'with-cuda'
Requires-Dist: nvidia-nvjitlink-cu12 >=12.1.105 ; extra == 'with-cuda'
Requires-Dist: nvidia-cuda-nvrtc-cu12 >=12.1.55 ; extra == 'with-cuda'
Requires-Dist: nvidia-nvshmem-cu12 >=3.2.5 ; extra == 'with-cuda'

