einops>=0.6
torch
