einops>=0.6.1
torch
numpy<2
