torch>=2.0
tqdm
tensorboard
numpy<=1.26.4
