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
- equinox >=0.11.0
- grain >=0.2.0
- jax >=0.4.36
- jaxlib >=0.4.36
- numpy >=1.26
- optax >=0.2.3
- orbax-checkpoint >=0.6.0
- pytest-asyncio >=0.23
19 optional dependencies
-
beartype>=0.18.0[dev]extra == "dev"
-
chex>=0.1.86[dev]extra == "dev"
-
complexipy>=5.6.1[dev]extra == "dev"
-
flatbuffers>=2[cli]extra == "cli"
-
import-linter>=2.1[dev]extra == "dev"
-
interrogate>=1.7.0[dev]extra == "dev"
-
jaxlint>=0.1.0a1[dev]extra == "dev"
-
jaxtyping>=0.2.28[dev]extra == "dev"
-
libcst>=1.5.0[dev]extra == "dev"
-
matplotlib>=3.8[eda]extra == "eda"
-
pandas>=2.0[eda]extra == "eda"
-
pytest>=8.0[dev]extra == "dev"
-
pytest-asyncio>=0.23[dev]extra == "dev"
-
pytest-benchmark>=4.0[dev]extra == "dev"
-
pytest-cov>=5.0[dev]extra == "dev"
-
ruff>=0.4.0[dev]extra == "dev"
-
seaborn>=0.13[eda]extra == "eda"
-
ty[dev]extra == "dev"
-
tyro<2,>=0.9[cli]extra == "cli"
Used By
- aminx 1.4K/mo