tqdm>=4.66.2
openai>=1.23.6
tiktoken>=0.6.0
transformers>=4.40.1
python-dotenv>=1.0.1
faiss-cpu>=1.8.0
ftfy>=6.2.0
dacite>=1.8.1
vllm>=0.4.0
pandas>=1.4.0

[all]
rank-llm[sglang]
rank-llm[genai]
rank-llm[pyserini]

[genai]
google-generativeai>=0.8.2

[pyserini]
pyserini==1.2.0

[sglang]
sglang[all]~=0.4.0

[tensorrt-llm]
tensorrt-llm[all]>=0.14.0

[training]
accelerate>=0.34.2
bitsandbytes>=0.44.1
datasets>=3.0.1
deepspeed>=0.15.1
six>=1.16.0
torch>=2.4.0
numpy>=1.26.4
