datasets<3.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

[bm25s]
bm25s>=0.2.6
PyStemmer>=2.2.0.3

[codecarbon]
codecarbon

[dev]
ruff==0.6.4
pytest
pytest-xdist
pytest-coverage

[flagembedding]
FlagEmbedding

[flash_attention]
flash-attn>=2.6.3

[gritlm]
gritlm>=1.0.2

[jina]
einops>=0.8.0

[leaderboard]
gradio>=5.7.1
gradio_rangeslider>=0.0.8

[model2vec]
model2vec>=0.3.0

[openai]
openai>=1.41.0
tiktoken>=0.8.0

[peft]
peft>=0.11.0

[pylate]
pylate>=1.1.4

[speedtask]
GPUtil>=1.4.0
psutil>=5.9.8

[xformers]
xformers>=0.0.29
