Package Insights
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100Weekly Downloads
GitHub Stars
Downloads by OS
Python Versions
Top Countries
Dependencies
- jaxlib <=0.10.0,>=0.10.0
- ml-dtypes >=0.5.0
- numpy >=2.0
- opt-einsum
- scipy >=1.14
19 optional dependencies
- jax-cuda12-plugin[cuda12]
- jax-cuda12-plugin[cuda]
- jax-cuda12-plugin[cuda12-local]
- jax-cuda13-plugin[cuda13]
- jax-cuda13-plugin[cuda13-local]
- jaxlib[rocm7-local]
- jaxlib[ci]
- jaxlib[cuda]
- jaxlib[cuda12]
- jaxlib[cuda12-local]
- jaxlib[cuda13]
- jaxlib[cuda13-local]
- jaxlib[minimum-jaxlib]
- jaxlib[tpu]
- jax-rocm7-plugin[rocm7-local]
- kubernetes[k8s]
- libtpu[tpu]
- requests[tpu]
- xprof[xprof]