requirements.txt

absl-py==2.1.0
accelerate==0.27.2
aiohttp 
aiosignal 
annotated-types==0.6.0
argon2-cffi==23.1.0
argon2-cffi-bindings==21.2.0
asttokens==2.4.1
astunparse==1.6.3
async-timeout 
attrs==23.2.0
audioread==3.0.1
beautifulsoup4==4.12.3
bleach==6.1.0
blis==0.7.11
cachetools==5.3.2
catalogue==2.0.10
certifi==2023.11.17
cffi==1.16.0
charset-normalizer==3.3.2
click==8.1.7
cloudpathlib==0.16.0
cloudpickle 
cmake==3.28.1
comm==0.2.1
confection==0.1.4
contourpy==1.2.0
# cubinlinker 
cuda-python 
cudf 
cugraph
cugraph-dgl 
cugraph-service-client
cugraph-service-server 
cuml 
cupy-cuda12x 
cycler==0.12.1
cymem==2.0.8
Cython==3.0.8
dask 
dask-cuda 
dask-cudf 
dataclasses-json==0.6.4
datasets==2.18.0
debugpy==1.8.0
decorator==5.1.1
defusedxml==0.7.1
dill==0.3.8
distributed 
dm-tree==0.1.8
einops==0.7.0
exceptiongroup==1.2.0
execnet==2.0.2
executing==2.0.1
expecttest==0.1.3
fastjsonschema==2.19.1
fastrlock 
filelock==3.13.1
flash-attn==2.0.4
fonttools==4.47.2
frozenlist 
gast==0.5.4
GEARLM==0.0.6
google-auth==2.26.2
google-auth-oauthlib==0.4.6
graphsurgeon 
grpcio==1.60.0
huggingface-hub==0.21.4
hypothesis==5.35.1
idna==3.6
importlib-metadata 
iniconfig==2.0.0
ipykernel==6.29.0
ipython==8.20.0
ipython-genutils==0.2.0
jedi==0.19.1
Jinja2==3.1.3
joblib==1.3.2
json5==0.9.14
jsonschema==4.21.1
jsonschema-specifications==2023.12.1
jupyter-tensorboard 
jupyter_client==8.6.0
jupyter_core==5.7.1
jupyterlab==2.3.2
jupyterlab-server==1.2.0
jupyterlab_pygments==0.3.0
jupytext==1.16.1
kiwisolver==1.4.5
langcodes==3.3.0
lazy_loader==0.3
librosa==0.10.1
llvmlite 
locket 
Markdown==3.5.2
markdown-it-py==3.0.0
MarkupSafe==2.1.4
marshmallow==3.21.1
matplotlib==3.8.2
matplotlib-inline==0.1.6
mdit-py-plugins==0.4.0
mdurl==0.1.2
mistune==3.0.2
mock==5.1.0
mpmath==1.3.0
msgpack==1.0.7
multidict 
multiprocess==0.70.16
murmurhash==1.0.10
mypy-extensions==1.0.0
nbclient==0.9.0
nbconvert==7.14.2
nbformat==5.9.2
nest-asyncio==1.5.9
networkx==2.6.3
ninja==1.11.1.1
notebook==6.4.10
numba 
numpy==1.24.4
nvfuser
nvidia-dali-cuda120==1.33.0
nvidia-pyindex==1.0.9
nvtx 
oauthlib==3.2.2
onnx 
opencv.python
optree==0.10.0
package-name==0.1
packaging==23.2
pandas 
pandocfilters==1.5.1
parso==0.8.3
partd 
pexpect==4.9.0
pillow 
platformdirs==4.1.0
pluggy==1.3.0
ply 
polygraphy
pooch==1.8.0
preshed==3.0.9
prettytable==3.9.0
prometheus-client==0.19.0
prompt-toolkit==3.0.43
protobuf==4.24.4
psutil 
ptxcompiler
ptyprocess==0.7.0
pure-eval==0.2.2
pyarrow 
pyarrow-hotfix==0.6
pyasn1==0.5.1
pyasn1-modules==0.3.0
pybind11==2.11.1
pybind11-global==2.11.1
pycocotools 
pycparser==2.21
pydantic==2.5.3
pydantic_core==2.14.6
Pygments==2.17.2
pylibcugraph
pylibcugraphops
pylibraft 
pynvml 
pyparsing==3.1.1
pytest==7.4.4
pytest-flakefinder==1.1.0
pytest-rerunfailures==13.0
pytest-shard==0.1.2
pytest-xdist==3.5.0
python-dateutil==2.8.2
python-hostlist==1.23.0
pytorch-quantization==2.1.2
pytz 
PyYAML==6.0.1
pyzmq==25.1.2
raft-dask 
rapids-dask-dependency 
referencing==0.32.1
regex==2023.12.25
requests==2.31.0
requests-oauthlib==1.3.1
rich 
rmm 
rpds-py==0.17.1
rsa==4.9
safetensors==0.4.2
scikit-learn 
scipy 
Send2Trash==1.8.2
sentencepiece==0.2.0
six==1.16.0
smart-open==6.4.0
sortedcontainers==2.4.0
soundfile==0.12.1
soupsieve==2.5
soxr==0.3.7
spacy==3.7.2
spacy-legacy==3.0.12
spacy-loggers==1.0.5
sphinx-glpi-theme==0.5
srsly==2.4.8
stack-data==0.6.3
sympy==1.12
tabulate==0.9.0
tblib 
tensorboard==2.9.0
tensorboard-data-server==0.6.1
tensorboard-plugin-wit==1.8.1
tensorrt 
terminado==0.18.0
thinc==8.2.2
threadpoolctl==3.2.0
thriftpy2 
tinycss2==1.2.1
tokenizers==0.15.2
toml==0.10.2
tomli==2.0.1
toolz 
torch 
torch-tensorrt 
torchdata 
torchtext 
torchvision 
tornado==6.4
tqdm==4.66.1
traitlets==5.9.0
transformer-engine 
transformers==4.38.2
treelite 
treelite-runtime 
typer==0.9.0
types-dataclasses==0.6.6
typing-inspect==0.9.0
typing_extensions==4.9.0
ucx-py 
uff 
urllib3 
wasabi==1.1.2
wcwidth==0.2.13
weasel==0.3.4
webencodings==0.5.1
Werkzeug==3.0.1
xdoctest==1.0.2
xgboost 
xxhash==3.4.1
yarl 
zict 
zipp