tqdm
torch>=1.12.1

[extras]
matplotlib
