The current version has a known security vulnerability. Review the advisory and consider upgrading.
Downloads in the 30 and 7 days to October 10, 2026. Refreshed weekly.
Package Information
Package Insights
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100More insights and metrics
Downloads by OS
Python Versions
Top Countries
Weekly Downloads
GitHub Stars *
Dependencies
-
cuda-bindings
<14,>=13.0.3
platform_system == "Linux" and python_version < "3.15"
-
cuda-toolkit
==13.0.3
platform_system == "Linux"
- filelock
- fsspec >=0.8.5
- jinja2
- networkx >=2.5.1
-
nvidia-cudnn-cu13
==9.24.0.43
platform_system == "Linux"
-
nvidia-cusparselt-cu13
==0.8.1
platform_system == "Linux"
-
nvidia-nccl-cu13
==2.30.7
platform_system == "Linux"
-
nvidia-nvshmem-cu13
==3.4.5
platform_system == "Linux"
- setuptools >=77.0.3
- sympy >=1.13.3
-
triton
~=3.8.0
platform_system == "Linux" and python_version < "3.15"
- typing-extensions >=4.10.0
3 optional dependencies
-
opt-einsum>=3.3[opt-einsum]extra == "opt-einsum"
-
optree>=0.13.0[optree]extra == "optree"
-
pyyaml[pyyaml]extra == "pyyaml"
Used By
- sentence-transformers 22.5M/mo
- accelerate 18.6M/mo
- torchvision 15.9M/mo
- torchmetrics 11.2M/mo
- timm 10.8M/mo
- pytorch-lightning 9.9M/mo
- peft 8.5M/mo
- openai-whisper 5.2M/mo
- xgrammar 4.6M/mo
- ultralytics 4.4M/mo