Package Insights
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100Weekly Downloads
GitHub Stars
Downloads by OS
Python Versions
Top Countries
Dependencies
- jax >=0.10.0
- msgpack
- numpy >=1.23.2
- optax
- orbax-checkpoint
- pyyaml >=5.4.1
- rich >=11.1
- tensorstore
- treescope >=0.1.7
- typing-extensions >=4.2
48 optional dependencies
- ale-py[testing]
- cloudpickle[testing]
- clu[testing]
- dm-haiku[docs]
- docutils[docs]
- einops[testing]
- einops[docs]
- grain[testing]
- gymnasium[testing]
- ipykernel[docs]
- ipython-genutils[docs]
- ipywidgets[docs]
- jaxlib[testing]
- jaxtyping[testing]
- jraph[testing]
- jupytext[docs]
- kagglehub[docs]
- keras[testing]
- matplotlib[docs]
- ml-collections[docs]
- ml-collections[testing]
- mypy[testing]
- myst-nb[docs]
- nanobind[dev]
- nbstripout[docs]
- opencv-python[testing]
- orbax-export[docs]
- pre-commit[dev]
- protobuf[testing]
- pygments[docs]
- pytest[testing]
- pytest-cov[testing]
- pytest-custom-exit-code[testing]
- pytest-xdist[testing]
- pytype[testing]
- recommonmark[docs]
- scikit-build-core[dev]
- scikit-learn[docs]
- sentencepiece[testing]
- sphinx[docs]
- sphinx-book-theme[docs]
- sphinx-design[docs]
- tensorflow[testing]
- tensorflow-datasets[testing]
- tensorflow-text[testing]
- torch[testing]
- tqdm[docs]
- treescope[testing]
Used By
- tensorflowjs
- clu
- aqtp
- qwix
- jax-md
- tpu-inference
- google-tunix
- gymnax
- brax
- dopamine-rl
- jax2onnx
- kauldron
- google-metrax
- praxis
- gemma
- paxml
- evosax
- xminigrid
- jax-ai-stack
- netket
- playground
- ximinf
- fusion-surrogates
- mplang-nightly
- easydel
- transformer-engine-jax
- pertpy
- flashbax
- jaxmarl
- alberta-framework
- dantinox
- craftax
- datarax
- qdax
- jax-tpu-embedding
- fjformer
- navix
- calibrax
- google-jetstream
- sbx-rl
- pycse
- cax
- avitai-artifex
- tojax
- metalog-jax
- safenax
- opifex
- sinapsis-huggingface-transformers
- evojax
- metrx