| # Python 3.12 package environment captured from the ColVec1.1 SDPA evaluation. |
| # Platform: Linux x86-64, PyTorch 2.9.0, CUDA 12.8. |
| # Install PyTorch from the CUDA 12.8 index before installing this file. |
| # This is the complete evaluation environment, not the minimal inference setup. |
|
|
| absl-py==2.5.0 |
| accelerate==1.14.0 |
| aiohappyeyeballs==2.7.1 |
| aiohttp==3.14.3 |
| aiohttp-cors==0.8.1 |
| aiosignal==1.4.0 |
| annotated-doc==0.0.4 |
| annotated-types==0.8.0 |
| antlr4-python3-runtime==4.9.3 |
| anyio==4.14.2 |
| apache-tvm-ffi==0.1.9 |
| attrs==26.1.0 |
| bitsandbytes==0.50.0 |
| boto3==1.43.56 |
| botocore==1.43.56 |
| braceexpand==0.1.7 |
| causal-conv1d @ https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.6.2.post1/causal_conv1d-1.6.2.post1+cu12torch2.9cxx11abiTRUE-cp312-cp312-linux_x86_64.whl |
| certifi==2026.7.22 |
| cffi==2.1.0 |
| charset-normalizer==3.4.9 |
| click==8.4.2 |
| cloudpickle==3.1.2 |
| colorful==0.5.8 |
| colpali-engine @ git+https://github.com/illuin-tech/colpali@2e0b927051af727238783af039dcc2c50a4d8c27 |
| contourpy==1.3.3 |
| cryptography==49.0.0 |
| cycler==0.12.1 |
| datasets==5.0.0 |
| dill==0.4.1 |
| distlib==0.4.3 |
| docstring-parser==0.18.0 |
| einops==0.8.2 |
| filelock==3.29.0 |
| fla-core==0.5.1 |
| flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3+cu12torch2.9cxx11abiTRUE-cp312-cp312-linux_x86_64.whl |
| flash-linear-attention==0.5.1 |
| fonttools==4.63.0 |
| frozenlist==1.8.0 |
| fsspec==2026.4.0 |
| gapstatistics==0.1.5 |
| google-api-core==2.33.0 |
| google-auth==2.56.2 |
| googleapis-common-protos==1.75.0 |
| grpcio==1.83.0 |
| h11==0.16.0 |
| h2==4.4.0 |
| hf-xet==1.5.2 |
| hpack==4.2.0 |
| httpcore==1.0.9 |
| httpx==0.28.1 |
| huggingface-hub==1.24.0 |
| hydra-core==1.3.4 |
| hyperframe==6.1.0 |
| idna==3.18 |
| importlib-resources==7.1.0 |
| iniconfig==2.3.0 |
| jinja2==3.1.6 |
| jmespath==1.1.0 |
| joblib==1.5.3 |
| jsonargparse==4.50.0 |
| jsonnet==0.22.0 |
| jsonschema==4.26.0 |
| jsonschema-specifications==2025.9.1 |
| kiwisolver==1.5.0 |
| lightning==2.6.5 |
| lightning-utilities==0.15.3 |
| markdown==3.10.2 |
| markdown-it-py==4.2.0 |
| markupsafe==3.0.3 |
| matplotlib==3.11.1 |
| mdurl==0.1.2 |
| ml-dtypes==0.5.4 |
| mpmath==1.3.0 |
| msgpack==1.2.1 |
| mteb[xet] @ git+https://github.com/embeddings-benchmark/mteb.git@d56a414b45ebad0d03495de000b4880d8b028d4a |
| multidict==6.7.1 |
| multiprocess==0.70.19 |
| narwhals==2.24.0 |
| networkx==3.6.1 |
| numpy==2.4.4 |
| nvidia-cublas-cu12==12.8.4.1 |
| nvidia-cuda-cupti-cu12==12.8.90 |
| nvidia-cuda-nvrtc-cu12==12.8.93 |
| nvidia-cuda-runtime-cu12==12.8.90 |
| nvidia-cudnn-cu12==9.10.2.21 |
| nvidia-cufft-cu12==11.3.3.83 |
| nvidia-cufile-cu12==1.13.1.3 |
| nvidia-curand-cu12==10.3.9.90 |
| nvidia-cusolver-cu12==11.7.3.90 |
| nvidia-cusparse-cu12==12.5.8.93 |
| nvidia-cusparselt-cu12==0.7.1 |
| nvidia-nccl-cu12==2.27.5 |
| nvidia-nvjitlink-cu12==12.8.93 |
| nvidia-nvshmem-cu12==3.3.20 |
| nvidia-nvtx-cu12==12.8.90 |
| omegaconf==2.3.1 |
| opencensus==0.11.4 |
| opencensus-context==0.1.3 |
| opentelemetry-api==1.44.0 |
| opentelemetry-exporter-prometheus==0.65b0 |
| opentelemetry-proto==1.44.0 |
| opentelemetry-sdk==1.44.0 |
| opentelemetry-semantic-conventions==0.65b0 |
| packaging==26.2 |
| pandas==3.0.5 |
| peft==0.19.1 |
| pillow==12.2.0 |
| platformdirs==4.11.0 |
| pluggy==1.6.0 |
| polars==1.43.0 |
| polars-runtime-32==1.43.0 |
| portalocker==3.2.0 |
| prometheus-client==0.26.0 |
| propcache==0.5.2 |
| proto-plus==1.28.2 |
| protobuf==7.35.1 |
| psutil==7.2.2 |
| py-spy==0.4.2 |
| pyarrow==25.0.0 |
| pyasn1==0.6.4 |
| pyasn1-modules==0.4.2 |
| pycparser==3.0 |
| pydantic==2.13.4 |
| pydantic-core==2.46.4 |
| pygments==2.20.0 |
| pymupdf==1.28.0 |
| pyparsing==3.3.2 |
| pytest==9.1.1 |
| python-dateutil==2.9.0.post0 |
| python-discovery==1.5.0 |
| python-dotenv==1.2.2 |
| pytorch-lightning==2.6.5 |
| pytrec-eval-terrier==0.5.10 |
| pyyaml==6.0.3 |
| qdrant-client==1.18.0 |
| ray==2.56.1 |
| referencing==0.37.0 |
| regex==2026.7.19 |
| requests==2.34.2 |
| rich==14.3.4 |
| rpds-py==2026.6.3 |
| s3transfer==0.19.2 |
| safetensors==0.8.0 |
| scikit-learn==1.9.0 |
| scipy==1.18.0 |
| sentence-transformers==5.6.0 |
| sentry-sdk==2.66.1 |
| setuptools==78.1.0 |
| shellingham==1.5.4 |
| six==1.17.0 |
| smart-open==8.0.1 |
| sympy==1.14.0 |
| tensorboard==2.21.0 |
| tensorboard-data-server==0.7.2 |
| tensorboardx==2.6.5 |
| threadpoolctl==3.6.0 |
| tilelang==0.1.9 |
| tokenizers==0.22.2 |
| torch==2.9.0+cu128 |
| torch-c-dlpack-ext==0.1.5 |
| torchmetrics==1.9.0 |
| torchvision==0.24.0+cu128 |
| tqdm==4.69.1 |
| transformers==5.14.1 |
| triton==3.5.0 |
| typer==0.27.0 |
| typeshed-client==2.12.0 |
| typing-extensions==4.15.0 |
| typing-inspection==0.4.2 |
| urllib3==2.7.0 |
| virtualenv==21.7.0 |
| wandb==0.28.1 |
| webdataset==1.0.2 |
| werkzeug==3.1.8 |
| wrapt==2.2.2 |
| xxhash==3.8.1 |
| yarl==1.24.5 |
| z3-solver==4.15.4.0 |
|
|
