requirements.txt

alembic==1.18.1
antlr4-python3-runtime==4.9.3
certifi==2026.1.4
charset-normalizer==3.4.4
colorlog==6.10.1
contourpy==1.3.2
cuda-bindings==12.9.4
cuda-pathfinder==1.3.3
cycler==0.12.1
deeprobust==0.2.11
dhg==0.9.5
filelock==3.20.3
fonttools==4.61.1
fsspec==2026.1.0
gensim==4.4.0
gmpy2==2.2.2
greenlet==3.3.0
hydra-core==1.3.2
idna==3.11
ImageIO==2.37.2
Jinja2==3.1.6
joblib==1.5.3
kiwisolver==1.4.9
lazy_loader==0.4
llvmlite==0.46.0
Mako==1.3.10
MarkupSafe==3.0.3
matplotlib==3.10.8
mpmath==1.3.0
networkx==3.4.2
numba==0.63.1
numpy==1.26.4
nvidia-cublas-cu11==11.10.3.66
nvidia-cublas-cu12==12.8.4.1
nvidia-cuda-cupti-cu12==12.8.90
nvidia-cuda-nvrtc-cu11==11.7.99
nvidia-cuda-nvrtc-cu12==12.8.93
nvidia-cuda-runtime-cu11==11.7.99
nvidia-cuda-runtime-cu12==12.8.90
nvidia-cudnn-cu11==8.5.0.96
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.4.5
nvidia-nvtx-cu12==12.8.90
omegaconf==2.3.0
optuna==4.7.0
packaging==26.0
pandas==2.3.3
patsy==1.0.2
pillow==12.1.0
protobuf==6.33.4
pyparsing==3.3.2
python-dateutil==2.9.0.post0
pytz==2025.2
PyYAML==6.0.3
requests==2.32.5
scikit-image==0.25.2
scikit-learn==1.7.2
scipy==1.15.3
seaborn==0.13.2
six==1.17.0
smart_open==7.5.0
SQLAlchemy==2.0.46
statsmodels==0.14.6
sympy==1.14.0
tensorboardX==2.6.4
texttable==1.7.0
threadpoolctl==3.6.0
tifffile==2025.5.10
tomli==2.4.0
torch==1.13.1
torchvision==0.14.1
tqdm==4.67.1
triton==3.6.0
typing_extensions==4.15.0
tzdata==2025.3
urllib3==2.6.3
wrapt==2.0.1