numpy
torch
ninja
requests
cuda-python
pynvml
einops
nvidia-nvshmem-cu12
