Package Insights
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100Weekly Downloads
GitHub Stars
Downloads by OS
Python Versions
Top Countries
Dependencies
- async-lru ~=2.0
- braceexpand >=0.1.7
- dataclasses-json ~=0.6.4
- datasets <5.0,>=3.1.0
- deepdiff
- draccus >=0.11.5
- equinox !=0.12.0,>=0.11.7
- filelock ~=3.13
- fsspec <2026,>=2024.2
- gcsfs <2026,>=2024.2
- haliax >=1.4.dev443
- humanfriendly ==10.0
- jax !=0.7.0,!=0.7.1,>=0.6.2
- jaxtyping >=0.2.34
- jinja2
- jmp >=0.0.3
- lenses
- matplotlib >=3.7.0
- optax >=0.1.9
- protobuf <7,>=6
- pyarrow >=11.0.0
- pydantic <3
- pytimeparse >=1.1.8
- ray >=2.34.0
- safetensors <0.7.0,>=0.4.2
- tblib <4.0.0,>=1.7.0
- tensorstore >=0.1.73
- tokenizers >=0.15.2
- tqdm-loggable >=0.2
- transformers <5.0,>=4.57.0
- wandb >=0.17.8
- zstandard >=0.18.0
15 optional dependencies
- fastapi[serve]
- google-api-python-client[gcp]
- google-auth[gcp]
- google-cloud-storage[gcp]
- google-cloud-storage-transfer[gcp]
- jax[tpu]
- jax[gpu]
- lm-eval[lm-eval]
- openai[serve]
- peft[torch-test]
- tensorboard[profiling]
- tensorboardx[profiling]
- torch[torch-test]
- uvicorn[serve]
- xprof[profiling]