torch>=2.8.0
pytorch-lightning>=2.3.3
numpy>=1.24.0
PyYAML>=6.0
matplotlib>=3.7.0
tqdm>=4.65.0
pydantic>=2.0
pandas>=2.0
jinja2>=3.0
joblib>=1.3
scipy>=1.10.0
POT>=0.9.0
scikit-learn>=1.3.0
huggingface_hub>=0.24.0
torchdiffeq>=0.2.3

[all]
seahorse-stpp[docs,hpo,runner]

[dev]
pytest>=8.0.0
ruff>=0.6.0
pre-commit>=3.7.0

[docs]
mkdocs-material>=9.5

[hpo]
ray[tune]>=2.43.0

[runner]
