jp6/cu126/: jax-cuda12-plugin-0.4.32.dev20240912 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.4.32.dev20240912
  • 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.1.105 ; 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.4.32.dev20240912-cp310-cp310-manylinux2014_aarch64.whl
Size
8 MB
Type
Python Wheel
Python
3.10

No description in metadata