numpy
matplotlib
numba
h5py
tqdm
rich
torch
torchvision
