numpy>=1.20
pyyaml>=6

[dev]
pytest
ruff

[torch]
torch==2.8.0
torch-geometric>=2.3.0
torch-scatter==2.1.2
torch-sparse==0.6.18
torch-cluster==1.6.3
torch-spline-conv==1.2.2
scipy>=1.9.0
transformers>=4.36.0
joblib>=1.3.0
scikit-learn>=1.3.0
ftfy>=6.1.0
regex>=2023.0.0
accelerate>=0.26.0
fschat>=0.2.36
