openteamsinc
pypi/jax-cuda12-plugin
Risk Profile
Package
Source
Risk Profile
Maturity:
Mature
Health:
High Risk
published package has a different name than specified in the source code
Legal:
Caution Needed
License includes a patent grant clause
License has additional text that may require review before use
Security:
Healthy
Package
pypi
jax-cuda12-plugin
Version
0.6.2
Last Release Date
30 days ago
License
Apache
Dependencies
jax-cuda12-pjrt
==0.6.2
with-cuda:
nvidia-cublas-cu12
>=12.1.3.1
nvidia-cuda-cupti-cu12
>=12.1.105
nvidia-cuda-nvcc-cu12
>=12.6.85
nvidia-cuda-runtime-cu12
>=12.1.105
nvidia-cudnn-cu12
<10.0, >=9.8
nvidia-cufft-cu12
>=11.0.2.54
nvidia-cusolver-cu12
>=11.4.5.107
nvidia-cusparse-cu12
>=12.1.0.106
nvidia-nccl-cu12
>=2.18.1
nvidia-nvjitlink-cu12
>=12.1.105
nvidia-cuda-nvrtc-cu12
>=12.1.55
nvidia-nvshmem-cu12
>=3.2.5
Source
Location
jax-ml/jax
Last Source Update
about 4 hours ago
Licenses
Apache License 2.0
Mozilla Public License 2.0
(text added)
BSD 3-Clause "New" or "Revised" License
BSD 3-Clause "New" or "Revised" License
MIT License
Distribution Destinations
pypi/jax-ffi-example
