tensorflow[and-cuda]
tf_keras
tf-models-official
npn
graphviz
nvidia-ml-py
bitarray
huggingface_hub
