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
- einops >=0.8.1
- kornia >=0.7.3
- natsort >=8.4.0
- omegaconf >=2.3.0
- pytorch-lightning >=2.4.0
- pytorch-optimizer >=3.4.2
- schedulefree >=1.4
- torch >=2.0
- torchgeometry >=0.1.2
- torchvision >=0.19.0
18 optional dependencies
-
codespell>=2.4.1[dev]extra == "dev"
-
distrax>=0.1.5[jax]extra == "jax"
-
einops>=0.8.1[jax]extra == "jax"
-
flax>=0.10.0[jax]extra == "jax"
-
ipython<=8.33.0[dev]extra == "dev"
-
jax>=0.4.30[jax]extra == "jax"
-
mkdocs>=1.6.1[docs]extra == "docs"
-
mkdocs-git-revision-date-localized-plugin>=1.4.7[docs]extra == "docs"
-
mkdocs-material>=9.7.1[docs]extra == "docs"
-
mkdocstrings>=0.26.1[docs]extra == "docs"
-
mypy>=1.15.0[dev]extra == "dev"
-
optax>=0.2.0[jax]extra == "jax"
-
pip>=25.3[dev]extra == "dev"
-
pre-commit>=3.5.0[dev]extra == "dev"
-
pytest>=8.3.5[dev]extra == "dev"
-
pytest-cov<=6.0.0[dev]extra == "dev"
-
ruff>=0.9.10[dev]extra == "dev"
-
types-pyserial>=3.5.0.20241221[dev]extra == "dev"