sentence-transformers==3.4.1
transformers>=4.34.0
optuna==4.2.0
wandb==0.19.4
torch>=2.0.0
ninja==1.11.1.3
scikit-learn==1.6.1
pandas>=2.0.0
pandarallel==1.6.5
numpy>=1.26.4

[benchmark]
sqlalchemy==2.0.37
psycopg2-binary==2.9.9
pgvector==0.3.6

[data]
fastparquet==2024.11.0
rdkit==2024.9.4
opencv-python==4.11.0.86
beautifulsoup4==4.12.3
fickling==0.1.3

[dev]
pytest
pytest-cov
pytest-xdist
ipykernel
ipywidgets
