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
- flax
- jax
- jaxtyping
- numpy
- orbax-export
- recurrentgemma
- typeguard ==2.13.3
12 optional dependencies
-
absl-py[dev]extra == "dev"
-
attrs[tensorflow]extra == "tensorflow"
-
chex[dev]extra == "dev"
-
orbax[dev]extra == "dev"
-
pyink[dev]extra == "dev"
-
pylint>=2.6.0[dev]extra == "dev"
-
pytest[dev]extra == "dev"
-
pytest-xdist[dev]extra == "dev"
-
scipy[tensorflow]extra == "tensorflow"
-
tensorflow[dev]extra == "dev"
-
tensorflow[tensorflow]extra == "tensorflow"
-
tf-keras[tensorflow]extra == "tensorflow"