einops>=0.7.0
einx[torch]>=0.1.3
torch
