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 >=2.3.0
- einshape
- immutabledict
- jax >=0.11.0
- jaxlib >=0.11.0
- jaxtyping >=0.3
- pydantic >=2.11.0
- qwix >=0.1.2
- tensorboardx
- tqdm
- xprof >=2.23.1
16 optional dependencies
-
chex>=0.1.91[test]extra == "test"
-
cuequivariance-jax>=0.10.0[cuda]extra == "cuda"
-
cuequivariance-ops-jax-cu13>=0.10.0[cuda]extra == "cuda"
-
flatbuffers[test]extra == "test"
-
flax[test]extra == "test"
-
google-benchmark>=1.9.0[bench]extra == "bench"
-
hypothesis[test]extra == "test"
-
jax==0.11.2[bench]extra == "bench"
-
jax>=0.11.0[cuda]extra == "cuda"
-
jax>=0.11.0[tpu]extra == "tpu"
-
jax-triton[cuda]extra == "cuda"
-
nvidia-cudnn-cu13>=9.23.0.39[cuda]extra == "cuda"
-
nvidia-cudnn-frontend>=1.28.0[cuda]extra == "cuda"
-
pytest[test]extra == "test"
-
pytest-xdist[test]extra == "test"
-
triton[cuda]extra == "cuda"
Used By
- tpu-inference 39.0K/mo
- hint-ppi 7.5K/mo
- fast-hint-ppi 2.3K/mo
- jax-image-models 1.6K/mo