Metadata-Version: 2.2
Name: jax-cuda12-plugin
Version: 0.5.3
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 :: 3 - Alpha
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Requires-Python: >=3.10
Description-Content-Type: text/markdown
Requires-Dist: jax-cuda12-pjrt==0.5.3
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.1; 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"
Dynamic: author
Dynamic: author-email
Dynamic: classifier
Dynamic: description-content-type
Dynamic: home-page
Dynamic: license
Dynamic: provides-extra
Dynamic: requires-dist
Dynamic: requires-python
Dynamic: summary
