numpy
pynvml
rich
torch
