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,>=1
- einops <1,>=0
- jaxtyping <1,>=0
- numpy <2.4.0,>=1.26.0
- sentencepiece >=0.2.0
- typeguard <3,>=2
5 optional dependencies
-
flax>=0.12.0[full]extra == "full" or extra == "jax" or extra == "test"
-
jax>=0.9.0[full]extra == "full" or extra == "jax" or extra == "test"
-
orbax-checkpoint>=0.1.7[full]extra == "full" or extra == "jax" or extra == "test"
-
pytest<9,>=8[test]extra == "test"
-
torch<3,>=2[full]extra == "full" or extra == "torch" or extra == "test"
Used By
- magenta-rt 749/mo
- sequence-layers 118/mo