torch
torchvision
optuna
matplotlib

[bench]
pytest-benchmark==4.0.0

[dev]
pre-commit==3.5.0
mypy==1.10.0
ruff==0.1.7
pytest==7.4.4
pytest-cov==6.2.1
hypothesis==6.138.2

[docs]
sphinx==7.2.6
myst-parser==2.0.0
sphinx-autodoc-typehints==1.23.0
