datasets
transformers
torch==2.8.0
elasticsearch==8.18.1
aiohttp
ujson
scipy
faiss-gpu-cu12