tqdm
matplotlib
numpy==1.26.4

[gpu]
torch
numpy==1.26.4
