jax-cuda12-plugin has 1.4M monthly downloads and 36K GitHub stars. Version 0.10.2 supports Python 3.11+, first published in 2023.
It is actively maintained under a permissive license, last released in Jun 2026.
It is built on top of jax-cuda12-pjrt. It is used by 1 tracked package, including jax-ik.
GitHub topics include jax.
Package Insights
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100Weekly Downloads
GitHub Stars
Downloads by OS
Python Versions
Top Countries
Dependencies
- jax-cuda12-pjrt ==0.10.2
12 optional dependencies
- nvidia-cublas-cu12[with-cuda]
- nvidia-cuda-cupti-cu12[with-cuda]
- nvidia-cuda-nvcc-cu12[with-cuda]
- nvidia-cuda-nvrtc-cu12[with-cuda]
- nvidia-cuda-runtime-cu12[with-cuda]
- nvidia-cudnn-cu12[with-cuda]
- nvidia-cufft-cu12[with-cuda]
- nvidia-cusolver-cu12[with-cuda]
- nvidia-cusparse-cu12[with-cuda]
- nvidia-nccl-cu12[with-cuda]
- nvidia-nvjitlink-cu12[with-cuda]
- nvidia-nvshmem-cu12[with-cuda]