numpy
torch
ninja
requests
cuda-python
pynvml
einops
nvidia-nvshmem-cu12
nvidia-cudnn-cu12
nvidia-cudnn-frontend>=1.13.0
