certifi
charset-normalizer
cmake
colorama
cuda-bindings
cuda-pathfinder
cuda-python
fbgemm-gpu
filelock
fsspec
idna
importlib-metadata
iopath
jinja2
lightning-utilities
lit
markupsafe
mpmath
mypy-extensions
networkx
numpy
nvidia-cublas
nvidia-cublas-cu12
nvidia-cuda-cccl
nvidia-cuda-cccl-cu12
nvidia-cuda-cupti
nvidia-cuda-cupti-cu12
nvidia-cuda-nvrtc
nvidia-cuda-nvrtc-cu12
nvidia-cuda-runtime
nvidia-cuda-runtime-cu12
nvidia-cudnn-cu12
nvidia-cudnn-cu13
nvidia-cufft
nvidia-cufft-cu12
nvidia-cufile
nvidia-cufile-cu12
nvidia-curand
nvidia-curand-cu12
nvidia-cusolver
nvidia-cusolver-cu12
nvidia-cusparse
nvidia-cusparse-cu12
nvidia-cusparselt-cu12
nvidia-cusparselt-cu13
nvidia-nccl-cu12
nvidia-nccl-cu13
nvidia-nvjitlink
nvidia-nvjitlink-cu12
nvidia-nvshmem-cu12
nvidia-nvshmem-cu13
nvidia-nvtx
nvidia-nvtx-cu12
packaging
pillow
portalocker
pyelftools
pyre-extensions
pytorch-triton
pytorch-triton-rocm
pytorch-triton-xpu
requests
rocm
rocm-sdk-core
rocm-sdk-device-gfx1010
rocm-sdk-device-gfx1011
rocm-sdk-device-gfx1012
rocm-sdk-device-gfx1030
rocm-sdk-device-gfx1031
rocm-sdk-device-gfx1032
rocm-sdk-device-gfx1033
rocm-sdk-device-gfx1034
rocm-sdk-device-gfx1035
rocm-sdk-device-gfx1036
rocm-sdk-device-gfx1100
rocm-sdk-device-gfx1101
rocm-sdk-device-gfx1102
rocm-sdk-device-gfx1103
rocm-sdk-device-gfx1150
rocm-sdk-device-gfx1151
rocm-sdk-device-gfx1152
rocm-sdk-device-gfx1153
rocm-sdk-device-gfx1200
rocm-sdk-device-gfx1201
rocm-sdk-device-gfx1250
rocm-sdk-device-gfx908
rocm-sdk-device-gfx90a
rocm-sdk-device-gfx942
rocm-sdk-device-gfx950
rocm-sdk-libraries
setuptools
spmd-types
sympy
torch
torchaudio
torchcsprng
torchdata
torchmetrics
torchrec
torchtext
torchvision
tqdm
triton
triton-rocm
triton-xpu
typing-extensions
typing-inspect
urllib3
zipp