torch<3,>=2.3
lightning<3,>=2.4
numpy<3,>=1.21.2
mlflow<4,>=3.11.1
PyYAML<7,>=6

[all]
scipy
scikit-learn
matplotlib

[develop]
pre-commit
ruff
pytest
pytest-cov
pytest-randomly
pytest-benchmark
mypy
types-PyYAML
types-setuptools

[plot]
matplotlib
scipy

[sampling]
scipy

[sklearn]
scikit-learn
