jax-cuda13-plugin has 209K monthly downloads and 36K GitHub stars. Version 0.10.2 supports Python 3.11+, first published in 2025.
It is actively maintained under a permissive license, last released in Jun 2026.
It is built on top of jax-cuda13-pjrt.
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-cuda13-pjrt ==0.10.2
13 optional dependencies
- nvidia-cublas[with-cuda]
- nvidia-cuda-cupti[with-cuda]
- nvidia-cuda-nvcc[with-cuda]
- nvidia-cuda-nvrtc[with-cuda]
- nvidia-cuda-runtime[with-cuda]
- nvidia-cudnn-cu13[with-cuda]
- nvidia-cufft[with-cuda]
- nvidia-cusolver[with-cuda]
- nvidia-cusparse[with-cuda]
- nvidia-nccl-cu13[with-cuda]
- nvidia-nvjitlink[with-cuda]
- nvidia-nvshmem-cu13[with-cuda]
- nvidia-nvvm[with-cuda]