torch
pycuda
triton
pynvml
