tqdm
matplotlib

[gpu]
cupy-cuda12x
numpy==1.26.4
