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]
- jax-rocm7-plugin[rocm7-local]
- jaxlib[minimum-jaxlib]
- jaxlib[ci]
- jaxlib[cuda]
- jaxlib[cuda12]
- jaxlib[cuda12-local]
- jaxlib[cuda13]
- jaxlib[cuda13-local]
- jaxlib[rocm7-local]
- jaxlib[tpu]
- kubernetes[k8s]
- libtpu[tpu]
- requests[tpu]
- xprof[xprof]
Used By
- orbax-checkpoint
- flax
- tensorflowjs
- optax
- chex
- equinox
- clu
- orbax-export
- aqtp
- e3nn-jax
- qwix
- jaxopt
- numpyro
- jax-md
- pathwaysutils
- nequix
- seqio
- drjax
- lineax
- tokamax
- tree-math
- tpu-inference
- jax-datetime
- seqio-nightly
- optimistix
- distrax
- diffrax
- gdsfactoryplus
- blackjax
- klujax
- kaggle-environments
- mujoco-mjx
- sax
- augmax
- jaxley
- jax-dataclasses
- tridiax
- jaxlie
- alphafold-colabfold
- gymnax
- interpax
- brax
- rax
- levanter
- sure-tools
- tensorflow-federated
- xdsl-jax
- jax-triton
- rlax
- dopamine-rl