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
- absl-py
- aiofiles
- etils
- humanize
- jax >=0.6.0
- msgpack
-
nest-asyncio
platform_system == "Windows"
- numpy
- prometheus-client >=0.20.0
- protobuf
- psutil
- pyyaml
- simplejson >=3.16.0
- tensorstore >=0.1.84
- typing-extensions
-
uvloop
platform_system != "Windows"
31 optional dependencies
-
aiofiles[docs]extra == "docs"
-
aiofiles[testing]extra == "testing"
-
aiosqlite[tiering-service]extra == "tiering-service"
-
chex[testing]extra == "testing"
-
clu[docs]extra == "docs"
-
clu[testing]extra == "testing"
-
fastapi[testing]extra == "testing"
-
fire[tiering-service]extra == "tiering-service"
-
flax[docs]extra == "docs"
-
flax[testing]extra == "testing"
-
google-cloud-logging[testing]extra == "testing"
-
google-cloud-logging[docs]extra == "docs"
-
grain[testing]extra == "testing"
-
grain[docs]extra == "docs"
-
greenlet[tiering-service]extra == "tiering-service"
-
grpcio-tools>=1.80.0[tiering-service]extra == "tiering-service"
-
httpx[testing]extra == "testing"
-
httpx[tiering-service]extra == "tiering-service"
-
mock[testing]extra == "testing"
-
opencv-python[docs]extra == "docs"
-
pysqlite3[tiering-service]extra == "tiering-service"
-
pytest[testing]extra == "testing"
-
pytest-xdist[testing]extra == "testing"
-
pytimeparse[tiering-service]extra == "tiering-service"
-
safetensors[docs]extra == "docs"
-
safetensors[testing]extra == "testing"
-
sqlalchemy>=1.4.0[tiering-service]extra == "tiering-service"
-
tensorflow[testing]extra == "testing"
-
tensorflow-datasets[docs]extra == "docs"
-
torch[testing]extra == "testing"
-
uvloop[tiering-service]extra == "tiering-service"
Used By
- flax 2.5M/mo
- pathwaysutils 394.2K/mo
- orbax-export 318.4K/mo
- google-tunix 263.0K/mo
- brax 41.1K/mo
- jax2onnx 20.9K/mo
- gemma 19.9K/mo
- substrax 19.7K/mo
- kauldron 16.5K/mo
- playground 12.2K/mo