new requirements

requirements.txt

@@ -1,3 +1,4 @@
+accelerate==0.30.1
 aiofiles==23.2.1
 aiohttp==3.9.5
 aiosignal==1.3.1
@@ -70,10 +71,24 @@ MarkupSafe==2.1.5
 marshmallow==3.21.2
 matplotlib==3.9.0
 mdurl==0.1.2
+mpmath==1.3.0
 multidict==6.0.5
 mypy-extensions==1.0.0
+networkx==3.3
 nltk==3.8.1
 numpy==1.26.4
+nvidia-cublas-cu12==12.1.3.1
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cudnn-cu12==8.9.2.26
+nvidia-cufft-cu12==11.0.2.54
+nvidia-curand-cu12==10.3.2.106
+nvidia-cusolver-cu12==11.4.5.107
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-nccl-cu12==2.20.5
+nvidia-nvjitlink-cu12==12.5.40
+nvidia-nvtx-cu12==12.1.105
 openai==1.30.5
 ordered-set==4.1.0
 orjson==3.10.3
@@ -85,6 +100,7 @@ platformdirs==4.2.2
 plumbum==1.8.3
 ply==3.11
 protobuf==4.25.3
+psutil==5.9.8
 pyarrow==16.1.0
 pydantic==2.7.2
 pydantic_core==2.18.3
@@ -117,6 +133,7 @@ sniffio==1.3.1
 soupsieve==2.5
 SQLAlchemy==2.0.30
 starlette==0.37.2
+sympy==1.12.1
 tabulate==0.9.0
 tenacity==8.3.0
 tiktoken==0.7.0
@@ -125,9 +142,11 @@ toml==0.10.2
 tomli==2.0.1
 tomlkit==0.12.0
 toolz==0.12.1
+torch==2.3.0
 tornado==6.4
 tqdm==4.66.4
 transformers==4.41.2
+triton==2.3.0
 typer==0.12.3
 typing-inspect==0.9.0
 typing_extensions==4.12.0