requirements.txt

astroid==2.15.5
click==8.1.3
cmake==3.26.3
contourpy==1.0.7
cycler==0.11.0
dill==0.3.6
docstring-to-markdown==0.12
filelock==3.12.0
fonttools==4.39.4
isort==5.12.0
jedi==0.18.2
Jinja2==3.1.2
kiwisolver==1.4.4
lazy-object-proxy==1.9.0
lit==16.0.5
MarkupSafe==2.1.2
matplotlib==3.7.1
mccabe==0.6.1
mpmath==1.3.0
mypy-extensions==1.0.0
networkx==3.1
numpy==1.24.3
nvidia-cublas-cu11==11.10.3.66
nvidia-cuda-cupti-cu11==11.7.101
nvidia-cuda-nvrtc-cu11==11.7.99
nvidia-cuda-runtime-cu11==11.7.99
nvidia-cudnn-cu11==8.5.0.96
nvidia-cufft-cu11==10.9.0.58
nvidia-curand-cu11==10.2.10.91
nvidia-cusolver-cu11==11.4.0.1
nvidia-cusparse-cu11==11.7.4.91
nvidia-nccl-cu11==2.14.3
nvidia-nvtx-cu11==11.7.91
packaging==23.1
parso==0.8.3
pathspec==0.11.1
Pillow==9.5.0
platformdirs==3.5.1
pluggy==1.0.0
pyflakes==2.4.0
pylint==2.17.4
pyparsing==3.0.9
python-dateutil==2.8.2
pytoolconfig==1.2.5
rope==1.8.0
six==1.16.0
snowballstemmer==2.2.0
sympy==1.12
tomli==2.0.1
tomlkit==0.11.8
torch==2.0.1
triton==2.0.0
typing_extensions==4.6.0
ujson==5.7.0
whatthepatch==1.0.5
wrapt==1.15.0
yapf==0.33.0