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.8.0,>=3.6.0
- einops
- pillow
- safetensors
- spmd-types ==0.2.3
- tensorboard
- tokenizers >=0.15.0
- torch-checkpointing >=0.1.0
- torchdata >=0.8.0
- tyro
- wandb
6 optional dependencies
-
expecttest[dev]extra == "dev"
-
pre-commit[dev]extra == "dev"
-
pyrefly==0.45.1[dev]extra == "dev"
-
pytest[dev]extra == "dev"
-
pytest-cov[dev]extra == "dev"
-
torch-remat>=0.2.0[remat]extra == "remat"