datasets<4.0.0,>=2.19.0
numpy<3.0.0,>=1.0.0
requests>=2.26.0
scikit_learn>=1.0.2
scipy>=0.0.0
sentence_transformers>=3.0.0
typing-extensions>=4.5.0
torch>1.0.0
tqdm>1.0.0
rich>=0.0.0
pytrec-eval-terrier>=0.5.6
pydantic>=2.0.0
typing_extensions>=0.0.0
eval_type_backport>=0.0.0
polars>=0.20.22

[ark]
volcengine-python-sdk[ark]==3.0.2
tiktoken>=0.8.0

[blip2]
salesforce-lavis>=1.0.2

[bm25s]
bm25s>=0.2.6
PyStemmer>=2.2.0.3

[codecarbon]
codecarbon<3.0.0,>=2.0.0

[cohere]
cohere==5.14.0

[colpali_engine]
colpali_engine>=0.3.12

[dev]
ruff==0.11.13
pytest>=8.3.4
pytest-xdist>=3.6.1
pytest-coverage>=0.0
pytest-rerunfailures>=15.0
iso639>=0.1.4
pre-commit>=4.1.0
bibtexparser>=1.4.3

[flagembedding]
FlagEmbedding==1.3.4

[flash_attention]
flash-attn>=2.6.3

[gritlm]
gritlm>=1.0.2

[image]
torchvision>0.2.1

[jina]
einops>=0.8.0

[jina-v4]
peft>=0.15.2
transformers>=4.52.0
torchvision>=0.22.1

[leaderboard]
plotly<6.0.0,>=5.24.0
cachetools>=5.2.0
matplotlib>=3.9.4

[leaderboard:python_version > "3.9"]
gradio==5.35.0

[llm2vec]
llm2vec<0.3.0,>=0.2.3

[model2vec]
model2vec>=0.3.0

[open_clip_torch]
open_clip_torch==2.31.0

[openai]
openai>=1.41.0
tiktoken>=0.8.0

[peft]
peft>=0.11.0

[pylate]
pylate>=1.1.4

[speedtask]
setuptools!=78.0.1
GPUtil>=1.4.0
psutil>=5.9.8

[timm]
timm<1.1.0,>=1.0.15

[vertexai]
vertexai==1.71.1

[voyage_v]
voyageai<2.0.0,>0.3.0
tenacity>9.0.0

[voyageai]
voyageai<2.0.0,>0.3.0

[xet]
huggingface_hub>=0.32.0

[xformers]
xformers>=0.0.29
