pandas>=1.3.5
numpy>=1.21.6, <2.0.0
scipy>=1.7.3,<=1.12.0
scikit-learn>=1.0.2
gensim
spacy
opencv-contrib-python>=4.9.0.80
tqdm
torch >= 2.0.0
torchvision >= 0.15.0
einops
wrapt
torchmetrics>=1.3.1
pyarrow>=15.0.0
fastparquet>=2024.2.0
transformers>=4.37.0
sentence-transformers>=2.3.0
sentencepiece>=0.2.0