torch>=2.0.0
numpy

[config]

[config:python_version < "3.11"]
tomli>=2.0.0

[dev]
pytest>=7.0
ruff>=0.14.13
mypy>=1.0.0
pre-commit>=3.0.0
matplotlib>=3.5.0
mkdocs>=1.5.0
mkdocs-material>=9.0.0
mkdocstrings[python]>=0.23.0

[tf]
tensorflow>=2.12
keras>=3.13.1

[viz]
matplotlib>=3.5.0
