certifi
charset-normalizer
cmake
colorama
cuda-bindings
cuda-pathfinder
cuda-python
cuda-toolkit
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
setuptools
spmd-types
sympy
torch
torch-tensorrt-rtx
torchaudio
torchcsprng
torchdata
torchmetrics
torchrec
torchtext
torchvision
tqdm
triton
triton-rocm
triton-xpu
typing-extensions
typing-inspect
urllib3
zipp