certifi
charset-normalizer
cmake
colorama
fbgemm-gpu
filelock
fsspec
idna
jinja2
lit
markupsafe
mpmath
networkx
numpy
nvidia-cublas-cu11
nvidia-cuda-cupti-cu11
nvidia-cuda-nvrtc-cu11
nvidia-cuda-runtime-cu11
nvidia-cudnn-cu11
nvidia-cufft-cu11
nvidia-curand-cu11
nvidia-cusolver-cu11
nvidia-cusparse-cu11
nvidia-nccl-cu11
nvidia-nvtx-cu11
packaging
pillow
pytorch-triton-rocm
requests
sympy
torch
torch-cuda80
torch-model-archiver
torch-tb-profiler
torch-tensorrt
torchaudio
torchcsprng
torchdata
torchmetrics
torchrec
torchrec-cpu
torchserve
torchtext
torchtune
torchvision
tqdm
triton
typing-extensions
urllib3
xformers