torch>=2.0
numpy
scikit-learn
matplotlib
scipy
dscribe
ase
pyyaml
optuna

[dev]
pytest
ruff
black
mypy
pre-commit
