einops>=0.8.1
torch>=2.7.0

[examples]
matplotlib
