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
- opencv-python-headless ==4.10.0.84
- pybind11
- pyyaml
- tiktoken
- torch <=2.10.0,>=2.6.0
- torchao
- torchdata
- transformers ==5.5.0
- wandb
34 optional dependencies
- albumentations[vlm]
- backoff[vlm]
- bitsandbytes[cuda-source]
- causal-conv1d[cuda]
- databricks-sql-connector[delta-databricks]
- deep-ep[moe]
- deltalake[delta-databricks]
- diffusers[diffusion]
- flash-attn[fa]
- ftfy[diffusion]
- imageio[diffusion]
- imageio-ffmpeg[diffusion]
- kernels[diffusion-kernels]
- mamba-ssm[cuda]
- mistral-common[vlm]
- nemo-automodel[all]
- nemo-automodel[moe]
- nemo-automodel[diffusion-kernels]
- numba[vlm]
- numpy[vlm]
- nv-grouped-gemm[cuda]
- onnxscript[cuda]
- open-clip-torch[vlm]
- opencv-python-headless[diffusion]
- perceptron[extra]
- pillow[vlm]
- pyyaml[cli]
- qwen-omni-utils[vlm]
- qwen-vl-utils[vlm]
- sentencepiece[extra]
- timm[vlm]
- torchcodec[vlm]
- torchvision[diffusion]
- transformer-engine[cuda]