nvidia-ml-py
torch
triton
