tqdm
matplotlib
numpy==1.26.4

[gpu]
cupy-cuda12x
numpy==1.26.4
