Downloads in the 30 and 7 days to October 10, 2026. Refreshed weekly.
Package Information
No reported malware or high/critical advisories · Details
The current version has no OSV.dev malware reports and no critical or high-severity vulnerabilities.
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
- einops
- hjson
- msgpack
- ninja
- numpy
- packaging >=20.0
- psutil
- py-cpuinfo
- pydantic >=2.0.0
- torch >=2.0.0
- tqdm
91 optional dependencies
-
accelerate[all]extra == "all"
-
accelerate[dev]extra == "dev"
-
autodoc-pydantic>=2.0.0[all]extra == "all"
-
autodoc-pydantic>=2.0.0[readthedocs]extra == "readthedocs"
-
clang-format==18.1.3[all]extra == "all"
-
clang-format==18.1.3[dev]extra == "dev"
-
comet-ml>=3.41.0[all]extra == "all"
-
comet-ml>=3.41.0[dev]extra == "dev"
-
deepspeed-kernels[all]sys_platform == "linux" and extra == "all"
-
deepspeed-kernels[dev]sys_platform == "linux" and extra == "dev"
-
diffusers>=0.25.0[sd]extra == "sd"
-
diffusers>=0.25.0[all]extra == "all"
-
docutils<0.18[readthedocs]extra == "readthedocs"
-
docutils<0.18[dev]extra == "dev"
-
docutils<0.18[all]extra == "all"
-
future[dev]extra == "dev"
-
future[all]extra == "all"
-
google[all]extra == "all"
-
google[inf]extra == "inf"
-
hjson[all]extra == "all"
-
hjson[readthedocs]extra == "readthedocs"
-
hjson[autotuning-ml]extra == "autotuning-ml"
-
importlib-metadata>=4[dev]extra == "dev"
-
importlib-metadata>=4[all]extra == "all"
-
lm-eval==0.3.0[all]extra == "all"
-
lm-eval==0.3.0[inf]extra == "inf"
-
mpi4py[all]extra == "all"
-
mpi4py[1bit-mpi]extra == "1bit-mpi"
-
mup[dev]extra == "dev"
-
mup[all]extra == "all"
-
neural-compressor==2.1.0[all]extra == "all"
-
neural-compressor==2.1.0[sparse]extra == "sparse"
-
packaging[all]extra == "all"
-
packaging[readthedocs]extra == "readthedocs"
-
pre-commit>=3.2.0[all]extra == "all"
-
pre-commit>=3.2.0[dev]extra == "dev"
-
protobuf[inf]extra == "inf"
-
protobuf[all]extra == "all"
-
psutil[readthedocs]extra == "readthedocs"
-
psutil[all]extra == "all"
-
py-cpuinfo[all]extra == "all"
-
py-cpuinfo[readthedocs]extra == "readthedocs"
-
pydantic>=2.0.0[readthedocs]extra == "readthedocs"
-
pydantic>=2.0.0[all]extra == "all"
-
pytest<8.4.0,>=7.2.0[dev]extra == "dev"
-
pytest<8.4.0,>=7.2.0[all]extra == "all"
-
pytest-forked[dev]extra == "dev"
-
pytest-forked[all]extra == "all"
-
pytest-randomly[all]extra == "all"
-
pytest-randomly[dev]extra == "dev"
-
pytest-xdist[all]extra == "all"
-
pytest-xdist[dev]extra == "dev"
-
qtorch[inf]extra == "inf"
-
qtorch[all]extra == "all"
-
qtorch==0.3.0[dev]extra == "dev"
-
recommonmark[readthedocs]extra == "readthedocs"
-
recommonmark[dev]extra == "dev"
-
recommonmark[all]extra == "all"
-
safetensors[inf]extra == "inf"
-
safetensors[all]extra == "all"
-
scipy[all]extra == "all"
-
scipy[deepcompile]extra == "deepcompile"
-
sentencepiece[inf]extra == "inf"
-
sentencepiece[all]extra == "all"
-
sphinx[dev]extra == "dev"
-
sphinx[all]extra == "all"
-
sphinx-rtd-theme[dev]extra == "dev"
-
sphinx-rtd-theme[all]extra == "all"
-
sphinx-rtd-theme[readthedocs]extra == "readthedocs"
-
tabulate[all]extra == "all"
-
tabulate[autotuning]extra == "autotuning"
-
tabulate[autotuning-ml]extra == "autotuning-ml"
-
tensorboard[dev]extra == "dev"
-
tensorboard[all]extra == "all"
-
torch>=2.0.0[all]extra == "all"
-
torch>=2.0.0[readthedocs]extra == "readthedocs"
-
torchembed>=0.3.1[torchembed]extra == "torchembed"
-
torchembed>=0.3.1[all]extra == "all"
-
torchvision[all]extra == "all"
-
torchvision[dev]extra == "dev"
-
tqdm[all]extra == "all"
-
tqdm[readthedocs]extra == "readthedocs"
-
transformers>=4.51.3[dev]extra == "dev"
-
transformers>=4.32.1[all]extra == "all"
-
transformers>=4.32.1[inf]extra == "inf"
-
triton>=2.1.0[all]extra == "all"
-
triton>=2.1.0[sd]extra == "sd"
-
wandb[all]extra == "all"
-
wandb[dev]extra == "dev"
-
xgboost[all]extra == "all"
-
xgboost[autotuning-ml]extra == "autotuning-ml"
Used By
- resemble-enhance 7.5K/mo
- hpsv3 4.3K/mo
- openrlhf 3.7K/mo
- protenix 3.6K/mo
- project-llm-trainer 3.0K/mo
- factory-sdk 1.9K/mo
- alphafold3-pytorch-lightning-hydra 1.0K/mo
- opentau 968/mo
- arbor-ai 939/mo
- trill-proteins 815/mo