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
- colorlog <7,>=6.10
- flax ==0.12.9
- h5py <4,>=3.16
- hydra-core ==1.3.6
- jax ==0.11.1
- mat73 <1,>=0.65
- numpy <3,>=2.4
- omegaconf ==2.3.1
- optax ==0.2.8
- orbax-checkpoint ==0.12.4
- pyyaml <7,>=6.0.3
- rich <16,>=15
- rootutils <2,>=1.0.7
- scipy <2,>=1.17
- tqdm <5,>=4.67
5 optional dependencies
-
jax==0.11.1[tpu]extra == "tpu"
-
jax==0.11.1[cuda13]extra == "cuda13"
-
jax==0.11.1[cuda12]extra == "cuda12"
-
tensorboard<3,>=2.20[tensorboard]extra == "tensorboard"
-
wandb<0.31,>=0.30[wandb]extra == "wandb"