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
- datasets >=4.0.0
- flashoptim >=0.1.3
- megatron-fsdp ==0.5.0
- mistral-common
- mlflow
- pybind11
- pyyaml
-
quack-kernels
==0.6.1
sys_platform == "linux"
- tiktoken
- torch >=2.6.0
- torchao
- torchdata
- transformers ==5.12.1
- wandb >=0.28.0
47 optional dependencies
-
albumentations[vlm-media]extra == "vlm-media"
-
apache-tvm-ffi<=0.1.11[cuda]extra == "cuda"
-
av[diffusion-media]extra == "diffusion-media"
-
backoff[vlm]extra == "vlm"
-
bitsandbytes[cuda-source]extra == "cuda-source"
-
boto3>=1.34[s3]extra == "s3"
-
causal-conv1d[cuda]extra == "cuda"
-
databricks-sql-connector>=4.4.0[delta-databricks]extra == "delta-databricks"
-
decord[vlm-media](platform_machine == "x86_64" and platform_system != "Darwin") and extra == "vlm-media"
-
deep-ep[moe]extra == "moe"
-
deltalake>=1.0.0[delta-databricks]extra == "delta-databricks"
-
diffusers>=0.39.0[diffusion]extra == "diffusion"
-
ffpa-attn>=0.2.2[ffpa]extra == "ffpa"
-
flash-attn<=2.8.3[fa]extra == "fa"
-
flash-linear-attention>=0.4.2[fla]extra == "fla"
-
ftfy[diffusion]extra == "diffusion"
-
imageio[diffusion]extra == "diffusion"
-
imageio-ffmpeg[diffusion-media]extra == "diffusion-media"
-
kernels[diffusion-kernels]extra == "diffusion-kernels"
-
mamba-ssm[cuda]extra == "cuda"
-
mistral-common>=1.11.0[vlm]extra == "vlm"
-
multi-storage-client>=0.13[msc]extra == "msc"
-
nemo-automodel[moe]extra == "moe"
-
nemo-automodel[all]extra == "all"
-
nemo-automodel[diffusion-kernels]extra == "diffusion-kernels"
-
nemo-automodel[media]extra == "media"
-
nemo-run[cli]extra == "cli"
-
numba[vlm]extra == "vlm"
-
numpy[vlm]extra == "vlm"
-
nv-grouped-gemm[cuda]extra == "cuda"
-
nvidia-cutlass-dsl==4.6.0[ffpa]extra == "ffpa"
-
onnxscript>=0.5.6[cuda]extra == "cuda"
-
open-clip-torch[vlm]extra == "vlm"
-
opencv-python-headless==4.10.0.84[vlm-media]extra == "vlm-media"
-
opencv-python-headless==4.10.0.84[diffusion-media]extra == "diffusion-media"
-
perceptron[extra]extra == "extra"
-
pillow[vlm]extra == "vlm"
-
pyyaml[cli]extra == "cli"
-
qwen-omni-utils[vlm-media]extra == "vlm-media"
-
qwen-vl-utils[vlm-media]extra == "vlm-media"
-
sentencepiece[extra]extra == "extra"
-
soundfile[vlm]extra == "vlm"
-
tile-kernels==1.0.0[cuda]extra == "cuda"
-
tilelang>=0.1.11[cuda]extra == "cuda"
-
timm<=1.0.22[vlm]extra == "vlm"
-
torchvision[diffusion]extra == "diffusion"
-
transformer-engine>=2.14.1[cuda]extra == "cuda"
Used By
- molt-rl 1.6K/mo