requirements-ml.txt

# LazyOwn optional machine-learning dependencies (pinned).
# Heavy, platform-specific (CUDA wheels on Linux x86-64). Code that uses these
# (modules/toposwarm_bridge.py, skills/lazyown_parquet_db.py) degrades gracefully
# when absent. Mirrors the pyproject.toml [project.optional-dependencies] ml extra.
# Install: pip install -r requirements-ml.txt   (or: pip install -e .[ml])

cuda-bindings==13.2.0
cuda-pathfinder==1.5.4
cuda-toolkit==13.0.2
joblib==1.5.3
nvidia-cublas==13.1.0.3
nvidia-cuda-cupti==13.0.85
nvidia-cuda-nvrtc==13.0.88
nvidia-cuda-runtime==13.0.96
nvidia-cudnn-cu13==9.19.0.56
nvidia-cufft==12.0.0.61
nvidia-cufile==1.15.1.6
nvidia-curand==10.4.0.35
nvidia-cusolver==12.0.4.66
nvidia-cusparse==12.6.3.3
nvidia-cusparselt-cu13==0.8.0
nvidia-nccl-cu13==2.28.9
nvidia-nvjitlink==13.0.88
nvidia-nvshmem-cu13==3.4.5
nvidia-nvtx==13.0.85
scikit-learn==1.8.0
scipy==1.17.1
seaborn==0.13.2
threadpoolctl==3.6.0
torch==2.12.1
torchaudio==2.12.1
torchvision==0.27.1
triton==3.6.0