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
- attrs >=21.3.0
- chex <0.1.81
- flax ==0.7.4
- importlab ==0.7
- jax ==0.4.21
- jaxlib ==0.4.21
- nltk ==3.7
- numpy <1.24
- optax ==0.1.7
- portpicker
- protobuf >=3.20.3
- seqio ==0.0.18
- tensorboard-plugin-profile ==2.14.0
- tensorflow ==2.14.0
- tensorflow-datasets >=4.9.2
- tensorflow-io >=0.34.0
-
tensorflow-text
==2.14.0
platform_machine == "x86_64"
- tensorstore >=0.1.21
- toml
- typing-extensions >=4.2.0
38 optional dependencies
-
apache-beam==2.52.0[dataflow]extra == "dataflow"
-
axlearn[tpu]extra == "tpu"
-
black==23.1a1[dev]extra == "dev"
-
cloud-tpu-client[gcp]extra == "gcp"
-
crcmod[gcp]extra == "gcp"
-
diffusers==0.16.1[dev]extra == "dev"
-
einops[dev]extra == "dev"
-
evaluate[dev]extra == "dev"
-
fairseq==0.12.2[dev]extra == "dev"
-
google-api-python-client==2.109.0[gcp]extra == "gcp"
-
google-apitools[dataflow]extra == "dataflow"
-
google-auth==2.23.0[gcp]extra == "gcp"
-
google-cloud-aiplatform[vertexai-tensorboard]extra == "vertexai-tensorboard"
-
google-cloud-core==2.3.3[gcp]extra == "gcp"
-
google-cloud-storage==2.11.0[gcp]extra == "gcp"
-
isort[dev]extra == "dev"
-
jax==0.4.21[tpu]extra == "tpu"
-
orjson==3.9.10[dataflow]extra == "dataflow"
-
pre-commit[dev]extra == "dev"
-
pycocotools[dev]extra == "dev"
-
pylint==2.13.9[dev]extra == "dev"
-
pyopenssl>=22.1.0[gcp]extra == "gcp"
-
pytest[dev]extra == "dev"
-
pytest-xdist[dev]extra == "dev"
-
pytype==2022.4.22[dev]extra == "dev"
-
scikit-learn>=1.1.1[dev]extra == "dev"
-
scipy[dev]extra == "dev"
-
sentencepiece!=0.1.92[dev]extra == "dev"
-
setuptools==65.7.0[vertexai-tensorboard]extra == "vertexai-tensorboard"
-
tensorboard[vertexai-tensorboard]extra == "vertexai-tensorboard"
-
tensorflow-metal==1.1.0[apple-silicon]extra == "apple-silicon" and platform_machine == "arm64"
-
timm==0.6.12[dev]extra == "dev"
-
torch>=1.12.1[dev]extra == "dev"
-
torchvision==0.16.1[dev]extra == "dev"
-
tqdm[dev]extra == "dev"
-
transformers==4.37.2[dev]extra == "dev"
-
wandb[dev]extra == "dev"
-
wrapt[dev]extra == "dev"