Package Insights
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100Weekly Downloads
GitHub Stars
Downloads by OS
Python Versions
Top Countries
Dependencies
- datasets >=4.0.0
- flashoptim >=0.1.3
- megatron-fsdp >=0.2.3
- mistral-common
- mlflow
- pybind11
- pyyaml
- tiktoken
- torch >=2.6.0
- torchao
- torchdata
- transformers ==5.8.1
- wandb >=0.28.0
43 optional dependencies
- apache-tvm-ffi[cuda]
- backoff[vlm]
- bitsandbytes[cuda-source]
- boto3[s3]
- causal-conv1d[cuda]
- databricks-sql-connector[delta-databricks]
- decord[vlm-media]
- deep-ep[moe]
- deltalake[delta-databricks]
- diffusers[diffusion]
- flash-attn[fa]
- flash-linear-attention[fla]
- ftfy[diffusion]
- imageio[diffusion]
- imageio-ffmpeg[diffusion-media]
- kernels[diffusion-kernels]
- mamba-ssm[cuda]
- mistral-common[vlm]
- multi-storage-client[msc]
- nemo-automodel[moe]
- nemo-automodel[all]
- nemo-automodel[media]
- nemo-automodel[diffusion-kernels]
- nemo-run[cli]
- numba[vlm]
- numpy[vlm]
- nv-grouped-gemm[cuda]
- onnxscript[cuda]
- open-clip-torch[vlm]
- opencv-python-headless[vlm-media]
- opencv-python-headless[diffusion-media]
- perceptron[extra]
- pillow[vlm]
- pyyaml[cli]
- qwen-omni-utils[vlm-media]
- qwen-vl-utils[vlm-media]
- sentencepiece[extra]
- soundfile[vlm]
- tile-kernels[cuda]
- tilelang[cuda]
- timm[vlm]
- torchvision[diffusion]
- transformer-engine[cuda]