einops>=0.8.0
einx>=0.3.0
loguru
packaging>=21.0
torch>=2.0
tqdm
matplotlib
numpy==1.26.4

[gpu]
torch
numpy==1.26.4
