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
- blobfile >=3.0.0
- datasets <=2.21.0,>=2.20.0
- einops >=0.8.1
- matplotlib >=3.7
- packaging <26.0,>=23.0
- psutil
- setuptools
- tiktoken >=0.9.0
- timm
- torchdata <1.0,>=0.8.0
- wandb
69 optional dependencies
-
apache-tvm-ffi==0.1.9[gpu]extra == "gpu"
-
av<15.0,>=14.3.0[npu]python_version == "3.11" and extra == "npu"
-
av<15.0,>=14.3.0[gpu](python_version == "3.11" and platform_machine == "x86_64") and extra == "gpu"
-
av<14.3,>=14.2.0[npu-aarch64]python_version == "3.11" and extra == "npu-aarch64"
-
create-block-mask-cuda==0.1[magi]sys_platform == "linux" and extra == "magi"
-
cuda-python==13.0.3[gpu]extra == "gpu"
-
debugpy[magi]extra == "magi"
-
decorator>=5.2.1[npu-aarch64]extra == "npu-aarch64"
-
decorator>=5.2.1[npu]extra == "npu"
-
diffusers==0.37.0[gpu]extra == "gpu"
-
diffusers==0.37.0[npu-aarch64]extra == "npu-aarch64"
-
diffusers==0.37.0[npu]extra == "npu"
-
expecttest<0.4,>=0.3.0[dev]extra == "dev"
-
flash-attn[gpu](sys_platform == "linux" and platform_machine == "x86_64") and extra == "gpu"
-
flash-attn-3[gpu](sys_platform == "linux" and (platform_machine == "aarch64" or platform_machine == "x86_64")) and extra == "gpu"
-
flash-attn-4==4.0.0b16[gpu]extra == "gpu"
-
flash-attn-cute==0.0.2[magi]sys_platform == "linux" and extra == "magi"
-
flash-linear-attention[gpu]extra == "gpu"
-
flash-mla[gpu](sys_platform == "linux" and (platform_machine == "aarch64" or platform_machine == "x86_64")) and extra == "gpu"
-
flash-qla==0.1.2[gpu]extra == "gpu"
-
ftfy[npu-aarch64]extra == "npu-aarch64"
-
ftfy[npu]extra == "npu"
-
ftfy[gpu]extra == "gpu"
-
gram-newton-schulz==0.1.6[gpu]extra == "gpu"
-
hf-transfer[gpu]extra == "gpu"
-
hf-transfer[npu]extra == "npu"
-
hf-transfer[npu-aarch64]extra == "npu-aarch64"
-
librosa<0.12,>=0.11.0[npu-aarch64]extra == "npu-aarch64"
-
librosa<0.12,>=0.11.0[npu]extra == "npu"
-
librosa<0.12,>=0.11.0[gpu]extra == "gpu"
-
liger-kernel[gpu]extra == "gpu"
-
magi-attention==1.1.1[magi]sys_platform == "linux" and extra == "magi"
-
magi-to-hstu-cuda==0.1[magi]sys_platform == "linux" and extra == "magi"
-
megatron-energon>=7.2.1[gpu]extra == "gpu"
-
megatron-energon>=7.2.1[npu]extra == "npu"
-
megatron-energon>=7.2.1[npu-aarch64]extra == "npu-aarch64"
-
nvidia-cudnn-frontend>=1.24.0[gpu]extra == "gpu"
-
nvidia-cusparselt-cu13[gpu]extra == "gpu"
-
nvidia-cutlass-dsl>=4.4.2[gpu]extra == "gpu"
-
nvidia-nccl-cu13[gpu]extra == "gpu"
-
peft==0.18.1[npu-aarch64]extra == "npu-aarch64"
-
peft==0.18.1[npu]extra == "npu"
-
peft==0.18.1[gpu]extra == "gpu"
-
pre-commit<5.0,>=4.0.0[dev]extra == "dev"
-
pytest<8.0,>=6.0.0[dev]extra == "dev"
-
quack-kernels==0.5.0[gpu]extra == "gpu"
-
ruff<1.0,>=0.7.0[dev]extra == "dev"
-
scipy>=1.16.2[npu]extra == "npu"
-
scipy>=1.16.2[npu-aarch64]extra == "npu-aarch64"
-
soundfile<0.14,>=0.13.1[gpu]extra == "gpu"
-
soundfile<0.14,>=0.13.1[npu]extra == "npu"
-
soundfile<0.14,>=0.13.1[npu-aarch64]extra == "npu-aarch64"
-
tile-kernels==1.0.0[gpu]extra == "gpu"
-
tilelang==0.1.9[gpu]extra == "gpu"
-
torch==2.10.0[npu-aarch64]extra == "npu-aarch64"
-
torch==2.11.0+cu130[gpu]extra == "gpu"
-
torch==2.10.0+cpu[npu]extra == "npu"
-
torch-c-dlpack-ext[gpu]extra == "gpu"
-
torch-npu==2.10.0[npu]extra == "npu"
-
torch-npu==2.10.0[npu-aarch64]extra == "npu-aarch64"
-
torchaudio==2.10.0[npu-aarch64]extra == "npu-aarch64"
-
torchaudio==2.11.0+cu130[gpu]extra == "gpu"
-
torchaudio==2.10.0+cpu[npu]extra == "npu"
-
torchcodec==0.12.0[gpu]extra == "gpu"
-
torchcodec==0.10.0[npu]extra == "npu"
-
torchvision==0.26.0+cu130[gpu]extra == "gpu"
-
torchvision==0.25.0+cpu[npu]extra == "npu"
-
torchvision==0.25.0[npu-aarch64]extra == "npu-aarch64"
-
veomni[magi]extra == "magi"