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
- einops
- filelock
- flax <0.13,>=0.12.10
- grain <0.3,>=0.2.18
- huggingface-hub
- jax <0.11.3,>=0.11.2
- jinja2
- numpy
- opencv-python-headless
- optax >=0.2.2
- orbax-checkpoint >=0.12.4
- pillow
- rich >=13
- transformers >=5.16.1
- typing-extensions >=4.14
- tyro
55 optional dependencies
-
compressed-tensors==0.17.1[test]extra == "test"
-
cosmic-ray==8.7.0[mutation]extra == "mutation"
-
datasets[test]extra == "test"
-
datasets[streaming]extra == "streaming"
-
diffusers>=0.34.0[diffusers]extra == "diffusers"
-
diffusers==0.34.0[test]extra == "test"
-
flatbuffers==25.12.19[test]extra == "test"
-
ftfy[wan]extra == "wan"
-
ftfy[test]extra == "test"
-
gguf==0.19.0[gguf]extra == "gguf"
-
ipykernel[test]extra == "test"
-
jax-cuda12-plugin<=0.11.2,>=0.11.2[cuda12]extra == "cuda12"
-
jax-cuda13-plugin<=0.11.2,>=0.11.2[cuda13]extra == "cuda13"
-
libtpu==0.0.48.*[tpu]extra == "tpu"
-
lm-eval>=0.4.10[eval-harness]extra == "eval-harness"
-
lm-eval==0.4.13[test]extra == "test"
-
matplotlib[plots]extra == "plots"
-
mlflow[test]extra == "test"
-
mlflow[mlflow]extra == "mlflow"
-
moviepy[av]extra == "av"
-
nbclient[test]extra == "test"
-
nbformat[test]extra == "test"
-
ollama<0.7,>=0.6.2[test]extra == "test"
-
ollama<0.7,>=0.6.2[inference-clients]extra == "inference-clients"
-
openai<4,>=3.8[inference-clients]extra == "inference-clients"
-
openai<4,>=3.8[test]extra == "test"
-
optuna[hpo]extra == "hpo"
-
optuna[test]extra == "test"
-
outlines-core>=0.2.14[test]extra == "test"
-
outlines-core>=0.2.14[guided]extra == "guided"
-
pytest[test]extra == "test"
-
pytest-xdist[test]extra == "test"
-
qwix==0.1.8[quantization]extra == "quantization"
-
requests[tpu]extra == "tpu"
-
requests[metrics]extra == "metrics"
-
ruff==0.14.3[test]extra == "test"
-
safetensors[interop]extra == "interop"
-
safetensors[test]extra == "test"
-
scipy[metrics]extra == "metrics"
-
scipy[test]extra == "test"
-
starlette>=0.40[serve]extra == "serve"
-
tensorboard[tensorboard]extra == "tensorboard"
-
tensorboard[test]extra == "test"
-
tensorflow-datasets>=4.9.10[tfds]extra == "tfds"
-
torch<2.15,>=2.14[torchax]extra == "torchax"
-
torch[torch]extra == "torch"
-
torch[test]extra == "test"
-
torch[diffusers]extra == "diffusers"
-
torchax==0.0.14.dev20260922[torchax]extra == "torchax"
-
torchvision>=0.29[vision]extra == "vision"
-
transformers==5.16.1[test]extra == "test"
-
uvicorn>=0.30[serve]extra == "serve"
-
wandb[test]extra == "test"
-
wandb[wandb]extra == "wandb"
-
xprof!=2.23.2,<3,>=2.23.1[profile]extra == "profile"
Used By
- sparxml 238/mo