jp6/cu128/: jax-cuda12-plugin-0.5.3 metadata and description

Homepage Simple index

JAX Plugin for NVIDIA GPUs

author JAX team
author_email jax-dev@google.com
classifiers
  • Development Status :: 3 - Alpha
  • Programming Language :: Python :: 3.10
  • Programming Language :: Python :: 3.11
  • Programming Language :: Python :: 3.12
description_content_type text/markdown
license Apache-2.0
requires_dist
  • jax-cuda12-pjrt==0.5.3
  • nvidia-cublas-cu12>=12.1.3.1; extra == "with-cuda"
  • nvidia-cuda-cupti-cu12>=12.1.105; extra == "with-cuda"
  • nvidia-cuda-nvcc-cu12>=12.6.85; extra == "with-cuda"
  • nvidia-cuda-runtime-cu12>=12.1.105; extra == "with-cuda"
  • nvidia-cudnn-cu12<10.0,>=9.1; extra == "with-cuda"
  • nvidia-cufft-cu12>=11.0.2.54; extra == "with-cuda"
  • nvidia-cusolver-cu12>=11.4.5.107; extra == "with-cuda"
  • nvidia-cusparse-cu12>=12.1.0.106; extra == "with-cuda"
  • nvidia-nccl-cu12>=2.18.1; extra == "with-cuda"
  • nvidia-nvjitlink-cu12>=12.1.105; extra == "with-cuda"
requires_python >=3.10

Because this project isn't in the mirror_whitelist, no releases from root/pypi are included.

File Tox results History
jax_cuda12_plugin-0.5.3-cp312-cp312-manylinux2014_aarch64.whl
Size
16 MB
Type
Python Wheel
Python
3.12

No description in metadata