nvidia-cupynumeric
torch
torchvision
nvtx
