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
- numpy >=1.26
-
typing-extensions
>=4.12
python_version < "3.12"
19 optional dependencies
-
flax>=0.10[jax]extra == "jax"
-
flax>=0.10[jax-gpu]extra == "jax-gpu"
-
flax>=0.10[all]extra == "all"
-
flax>=0.10[all-cpu]extra == "all-cpu"
-
jax>=0.4.30[jax]extra == "jax"
-
jax>=0.4.30[jax-gpu]extra == "jax-gpu"
-
jax>=0.4.30[keras]extra == "keras"
-
jax>=0.4.30[keras-gpu]extra == "keras-gpu"
-
jax>=0.4.30[all-cpu]extra == "all-cpu"
-
jax>=0.4.30[all]extra == "all"
-
keras>=3.5[keras]extra == "keras"
-
keras>=3.5[keras-gpu]extra == "keras-gpu"
-
keras>=3.5[all]extra == "all"
-
keras>=3.5[all-cpu]extra == "all-cpu"
-
torch>=2.4[all-cpu]extra == "all-cpu"
-
torch>=2.4[torch]extra == "torch"
-
torch>=2.4[torch-cpu]extra == "torch-cpu"
-
torch>=2.12[torch-gpu]extra == "torch-gpu"
-
torch>=2.4[all]extra == "all"