torch>=2.0
transformers>=4.35
numpy>=1.21
scikit-learn>=1.0
tqdm
torch-geometric>=2.4

[dev]
pytest
black
ruff
build
twine
